Spaces:
Paused
Paused
Update diffusion_webui/utils/preprocces_utils.py
Browse files
diffusion_webui/utils/preprocces_utils.py
CHANGED
@@ -72,6 +72,8 @@ def scribble_xdog(img, res=512, thr_a=32, **kwargs):
|
|
72 |
result[2 * (255 - dog) > thr_a] = 255
|
73 |
return remove_pad(result), True
|
74 |
|
|
|
|
|
75 |
|
76 |
PREPROCCES_DICT = {
|
77 |
"Hed": HEDdetector.from_pretrained("lllyasviel/Annotators"),
|
@@ -88,6 +90,7 @@ PREPROCCES_DICT = {
|
|
88 |
"Canny": CannyDetector(),
|
89 |
"ContentShuffle": ContentShuffleDetector(),
|
90 |
"MediapipeFace": MediapipeFaceDetector(),
|
91 |
-
"ScribbleXDOG": scribble_xdog
|
|
|
92 |
}
|
93 |
|
|
|
72 |
result[2 * (255 - dog) > thr_a] = 255
|
73 |
return remove_pad(result), True
|
74 |
|
75 |
+
def none_preprocces(image_path:str):
|
76 |
+
return Image.open(image_path)
|
77 |
|
78 |
PREPROCCES_DICT = {
|
79 |
"Hed": HEDdetector.from_pretrained("lllyasviel/Annotators"),
|
|
|
90 |
"Canny": CannyDetector(),
|
91 |
"ContentShuffle": ContentShuffleDetector(),
|
92 |
"MediapipeFace": MediapipeFaceDetector(),
|
93 |
+
"ScribbleXDOG": scribble_xdog,
|
94 |
+
"None": none_preprocces
|
95 |
}
|
96 |
|