Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: adaptation of Lineart Anime processor

Browse Source 
Adapted from https://github.com/huggingface/controlnet_aux
This commit is contained in:
psychedelicious
2024-03-21 20:06:14 +11:00
committed by Kent Keirsey
parent c7f8fe4d5e
commit c36d12a50f
2 changed files with 207 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
invokeai/app/invocations/controlnet_image_processors.py
View File

@ -9,7 +9,6 @@ import numpy as np
from controlnet_aux import (
ContentShuffleDetector,
LeresDetector,
LineartAnimeDetector,
MediapipeFaceDetector,
MidasDetector,
MLSDdetector,
@ -41,6 +40,7 @@ from invokeai.backend.image_util.depth_anything import DepthAnythingDetector
from invokeai.backend.image_util.dw_openpose import DWOpenposeDetector
from invokeai.backend.image_util.hed import HEDProcessor
from invokeai.backend.image_util.lineart import LineartProcessor
from invokeai.backend.image_util.lineart_anime import LineartAnimeProcessor
from .baseinvocation import BaseInvocation, BaseInvocationOutput, invocation, invocation_output
@ -264,9 +264,9 @@ class LineartAnimeImageProcessorInvocation(ImageProcessorInvocation):
detect_resolution: int = InputField(default=512, ge=0, description=FieldDescriptions.detect_res)
image_resolution: int = InputField(default=512, ge=0, description=FieldDescriptions.image_res)
def run_processor(self, image):
processor = LineartAnimeDetector.from_pretrained("lllyasviel/Annotators")
processed_image = processor(
def run_processor(self, image: Image.Image) -> Image.Image:
processor = LineartAnimeProcessor()
processed_image = processor.run(
image,
detect_resolution=self.detect_resolution,
image_resolution=self.image_resolution,