kadirnar commited on
Commit
920baa1
1 Parent(s): 860a671

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