Commit
·
d077fcd
1
Parent(s):
19dcf85
28. Sept. 2024, 12:45
Browse files
app.py
CHANGED
|
@@ -29,7 +29,7 @@ title = "ControlNet v1.1 Preprocessors Standalone"
|
|
| 29 |
|
| 30 |
# MARK: Funktionen
|
| 31 |
|
| 32 |
-
#
|
| 33 |
model_canny = None
|
| 34 |
def canny(img, res, l, h):
|
| 35 |
img = resize_image(HWC3(img), res)
|
|
@@ -41,7 +41,7 @@ def canny(img, res, l, h):
|
|
| 41 |
result = model_canny(img, l, h)
|
| 42 |
return [result]
|
| 43 |
|
| 44 |
-
#
|
| 45 |
model_hed = None
|
| 46 |
def hed(img, res):
|
| 47 |
img = resize_image(HWC3(img), res)
|
|
@@ -53,7 +53,7 @@ def hed(img, res):
|
|
| 53 |
result = model_hed(img)
|
| 54 |
return [result]
|
| 55 |
|
| 56 |
-
#
|
| 57 |
model_pidi = None
|
| 58 |
def pidi(img, res):
|
| 59 |
img = resize_image(HWC3(img), res)
|
|
@@ -65,7 +65,7 @@ def pidi(img, res):
|
|
| 65 |
result = model_pidi(img)
|
| 66 |
return [result]
|
| 67 |
|
| 68 |
-
#
|
| 69 |
model_mlsd = None
|
| 70 |
def mlsd(img, res, thr_v, thr_d):
|
| 71 |
img = resize_image(HWC3(img), res)
|
|
@@ -77,7 +77,7 @@ def mlsd(img, res, thr_v, thr_d):
|
|
| 77 |
result = model_mlsd(img, thr_v, thr_d)
|
| 78 |
return [result]
|
| 79 |
|
| 80 |
-
#
|
| 81 |
model_midas = None
|
| 82 |
def midas(img, res):
|
| 83 |
img = resize_image(HWC3(img), res)
|
|
@@ -89,7 +89,7 @@ def midas(img, res):
|
|
| 89 |
result = model_midas(img)
|
| 90 |
return [result]
|
| 91 |
|
| 92 |
-
#
|
| 93 |
model_zoe = None
|
| 94 |
def zoe(img, res):
|
| 95 |
img = resize_image(HWC3(img), res)
|
|
@@ -101,7 +101,7 @@ def zoe(img, res):
|
|
| 101 |
result = model_zoe(img)
|
| 102 |
return [result]
|
| 103 |
|
| 104 |
-
#
|
| 105 |
model_normalbae = None
|
| 106 |
def normalbae(img, res):
|
| 107 |
img = resize_image(HWC3(img), res)
|
|
@@ -113,7 +113,7 @@ def normalbae(img, res):
|
|
| 113 |
result = model_normalbae(img)
|
| 114 |
return [result]
|
| 115 |
|
| 116 |
-
#
|
| 117 |
model_dwpose = None
|
| 118 |
def dwpose(img, res):
|
| 119 |
img = resize_image(HWC3(img), res)
|
|
@@ -125,7 +125,7 @@ def dwpose(img, res):
|
|
| 125 |
result = model_dwpose(img)
|
| 126 |
return [result]
|
| 127 |
|
| 128 |
-
#
|
| 129 |
model_openpose = None
|
| 130 |
def openpose(img, res, hand_and_face):
|
| 131 |
img = resize_image(HWC3(img), res)
|
|
@@ -149,7 +149,7 @@ def openpose(img, res, hand_and_face):
|
|
| 149 |
# return [result]
|
| 150 |
|
| 151 |
|
| 152 |
-
#
|
| 153 |
model_lineart_anime = None
|
| 154 |
model_lineart = None
|
| 155 |
def lineart(img, res, preprocessor_name="Lineart", invert=True, old_images=None):
|
|
@@ -186,7 +186,7 @@ def lineart(img, res, preprocessor_name="Lineart", invert=True, old_images=None)
|
|
| 186 |
result_images.extend(old_images)
|
| 187 |
return result_images
|
| 188 |
|
| 189 |
-
#
|
| 190 |
model_oneformer_coco = None
|
| 191 |
def oneformer_coco(img, res):
|
| 192 |
img = resize_image(HWC3(img), res)
|
|
@@ -198,7 +198,7 @@ def oneformer_coco(img, res):
|
|
| 198 |
result = model_oneformer_coco(img)
|
| 199 |
return [result]
|
| 200 |
|
| 201 |
-
#
|
| 202 |
model_oneformer_ade20k = None
|
| 203 |
def oneformer_ade20k(img, res):
|
| 204 |
img = resize_image(HWC3(img), res)
|
|
@@ -210,7 +210,7 @@ def oneformer_ade20k(img, res):
|
|
| 210 |
result = model_oneformer_ade20k(img)
|
| 211 |
return [result]
|
| 212 |
|
| 213 |
-
#
|
| 214 |
model_content_shuffler = None
|
| 215 |
def content_shuffler(img, res):
|
| 216 |
img = resize_image(HWC3(img), res)
|
|
@@ -222,7 +222,7 @@ def content_shuffler(img, res):
|
|
| 222 |
result = model_content_shuffler(img)
|
| 223 |
return [result]
|
| 224 |
|
| 225 |
-
#
|
| 226 |
model_color_shuffler = None
|
| 227 |
def color_shuffler(img, res):
|
| 228 |
img = resize_image(HWC3(img), res)
|
|
@@ -234,7 +234,7 @@ def color_shuffler(img, res):
|
|
| 234 |
result = model_color_shuffler(img)
|
| 235 |
return [result]
|
| 236 |
|
| 237 |
-
#
|
| 238 |
model_inpaint = None
|
| 239 |
def inpaint(image, invert):
|
| 240 |
# color = HWC3(image["image"])
|
|
|
|
| 29 |
|
| 30 |
# MARK: Funktionen
|
| 31 |
|
| 32 |
+
# Canny
|
| 33 |
model_canny = None
|
| 34 |
def canny(img, res, l, h):
|
| 35 |
img = resize_image(HWC3(img), res)
|
|
|
|
| 41 |
result = model_canny(img, l, h)
|
| 42 |
return [result]
|
| 43 |
|
| 44 |
+
# Hed
|
| 45 |
model_hed = None
|
| 46 |
def hed(img, res):
|
| 47 |
img = resize_image(HWC3(img), res)
|
|
|
|
| 53 |
result = model_hed(img)
|
| 54 |
return [result]
|
| 55 |
|
| 56 |
+
# Pidi
|
| 57 |
model_pidi = None
|
| 58 |
def pidi(img, res):
|
| 59 |
img = resize_image(HWC3(img), res)
|
|
|
|
| 65 |
result = model_pidi(img)
|
| 66 |
return [result]
|
| 67 |
|
| 68 |
+
# MLSD
|
| 69 |
model_mlsd = None
|
| 70 |
def mlsd(img, res, thr_v, thr_d):
|
| 71 |
img = resize_image(HWC3(img), res)
|
|
|
|
| 77 |
result = model_mlsd(img, thr_v, thr_d)
|
| 78 |
return [result]
|
| 79 |
|
| 80 |
+
# Midas
|
| 81 |
model_midas = None
|
| 82 |
def midas(img, res):
|
| 83 |
img = resize_image(HWC3(img), res)
|
|
|
|
| 89 |
result = model_midas(img)
|
| 90 |
return [result]
|
| 91 |
|
| 92 |
+
# Zoe
|
| 93 |
model_zoe = None
|
| 94 |
def zoe(img, res):
|
| 95 |
img = resize_image(HWC3(img), res)
|
|
|
|
| 101 |
result = model_zoe(img)
|
| 102 |
return [result]
|
| 103 |
|
| 104 |
+
# Normal Bae
|
| 105 |
model_normalbae = None
|
| 106 |
def normalbae(img, res):
|
| 107 |
img = resize_image(HWC3(img), res)
|
|
|
|
| 113 |
result = model_normalbae(img)
|
| 114 |
return [result]
|
| 115 |
|
| 116 |
+
# DWPose
|
| 117 |
model_dwpose = None
|
| 118 |
def dwpose(img, res):
|
| 119 |
img = resize_image(HWC3(img), res)
|
|
|
|
| 125 |
result = model_dwpose(img)
|
| 126 |
return [result]
|
| 127 |
|
| 128 |
+
# OpenPose
|
| 129 |
model_openpose = None
|
| 130 |
def openpose(img, res, hand_and_face):
|
| 131 |
img = resize_image(HWC3(img), res)
|
|
|
|
| 149 |
# return [result]
|
| 150 |
|
| 151 |
|
| 152 |
+
# Lineart
|
| 153 |
model_lineart_anime = None
|
| 154 |
model_lineart = None
|
| 155 |
def lineart(img, res, preprocessor_name="Lineart", invert=True, old_images=None):
|
|
|
|
| 186 |
result_images.extend(old_images)
|
| 187 |
return result_images
|
| 188 |
|
| 189 |
+
# OneFormer
|
| 190 |
model_oneformer_coco = None
|
| 191 |
def oneformer_coco(img, res):
|
| 192 |
img = resize_image(HWC3(img), res)
|
|
|
|
| 198 |
result = model_oneformer_coco(img)
|
| 199 |
return [result]
|
| 200 |
|
| 201 |
+
# OneFormer ADE20k
|
| 202 |
model_oneformer_ade20k = None
|
| 203 |
def oneformer_ade20k(img, res):
|
| 204 |
img = resize_image(HWC3(img), res)
|
|
|
|
| 210 |
result = model_oneformer_ade20k(img)
|
| 211 |
return [result]
|
| 212 |
|
| 213 |
+
# Content Shuffler
|
| 214 |
model_content_shuffler = None
|
| 215 |
def content_shuffler(img, res):
|
| 216 |
img = resize_image(HWC3(img), res)
|
|
|
|
| 222 |
result = model_content_shuffler(img)
|
| 223 |
return [result]
|
| 224 |
|
| 225 |
+
# Color Shuffler
|
| 226 |
model_color_shuffler = None
|
| 227 |
def color_shuffler(img, res):
|
| 228 |
img = resize_image(HWC3(img), res)
|
|
|
|
| 234 |
result = model_color_shuffler(img)
|
| 235 |
return [result]
|
| 236 |
|
| 237 |
+
# Inpaint
|
| 238 |
model_inpaint = None
|
| 239 |
def inpaint(image, invert):
|
| 240 |
# color = HWC3(image["image"])
|