mirror of
https://github.com/lllyasviel/stable-diffusion-webui-forge.git
synced 2026-01-27 11:29:46 +00:00
49 lines
1.6 KiB
Python
49 lines
1.6 KiB
Python
import cv2
|
|
|
|
shared_preprocessors = {}
|
|
|
|
|
|
class PreprocessorParameter:
|
|
def __init__(self, minimum=0.0, maximum=1.0, step=0.01, label='Parameter 1', value=0.5, visible=False):
|
|
self.gradio_update_kwargs = dict(
|
|
minimum=minimum, maximum=maximum, step=step, label=label, value=value, visible=visible
|
|
)
|
|
|
|
|
|
class PreprocessorBase:
|
|
def __init__(self):
|
|
self.name = 'PreprocessorBase'
|
|
self.slider_1 = PreprocessorParameter()
|
|
self.slider_2 = PreprocessorParameter()
|
|
self.slider_3 = PreprocessorParameter()
|
|
|
|
def __call__(self, input_image, slider_1=None, slider_2=None, slider_3=None, **kwargs):
|
|
return input_image
|
|
|
|
|
|
class PreprocessorNone(PreprocessorBase):
|
|
def __init__(self):
|
|
super().__init__()
|
|
self.name = 'None'
|
|
|
|
def __call__(self, input_image, slider_1=None, slider_2=None, slider_3=None, **kwargs):
|
|
return input_image
|
|
|
|
|
|
shared_preprocessors['none'] = PreprocessorNone()
|
|
|
|
|
|
class PreprocessorCanny(PreprocessorBase):
|
|
def __init__(self):
|
|
super().__init__()
|
|
self.name = 'canny'
|
|
self.slider_1 = PreprocessorParameter(minimum=0, maximum=256, step=1, value=100, label='Low Threshold', visible=True)
|
|
self.slider_2 = PreprocessorParameter(minimum=0, maximum=256, step=1, value=200, label='High Threshold', visible=True)
|
|
|
|
def __call__(self, input_image, slider_1=100, slider_2=200, slider_3=None, **kwargs):
|
|
canny_image = cv2.cvtColor(cv2.Canny(input_image, int(slider_1), int(slider_2)), cv2.COLOR_GRAY2RGB)
|
|
return canny_image
|
|
|
|
|
|
shared_preprocessors['canny'] = PreprocessorCanny()
|