Spaces:
Running
on
L4
Running
on
L4
add pre_face_align option.
Browse files
app.py
CHANGED
|
@@ -108,15 +108,17 @@ def inference(image, face_align, background_enhance, face_upsample, upscale, cod
|
|
| 108 |
"""Run a single prediction on the model"""
|
| 109 |
try: # global try
|
| 110 |
# take the default setting for the demo
|
| 111 |
-
has_aligned = not face_align
|
| 112 |
only_center_face = False
|
| 113 |
draw_box = False
|
| 114 |
detection_model = "retinaface_resnet50"
|
| 115 |
-
print('Inp:', image, background_enhance, face_upsample, upscale, codeformer_fidelity)
|
| 116 |
|
|
|
|
|
|
|
| 117 |
background_enhance = background_enhance if background_enhance is not None else True
|
| 118 |
face_upsample = face_upsample if face_upsample is not None else True
|
| 119 |
upscale = upscale if (upscale is not None and upscale > 0) else 2
|
|
|
|
|
|
|
| 120 |
upscale = 1 if has_aligned else upscale
|
| 121 |
|
| 122 |
img = cv2.imread(str(image), cv2.IMREAD_COLOR)
|
|
|
|
| 108 |
"""Run a single prediction on the model"""
|
| 109 |
try: # global try
|
| 110 |
# take the default setting for the demo
|
|
|
|
| 111 |
only_center_face = False
|
| 112 |
draw_box = False
|
| 113 |
detection_model = "retinaface_resnet50"
|
|
|
|
| 114 |
|
| 115 |
+
print('Inp:', image, background_enhance, face_upsample, upscale, codeformer_fidelity)
|
| 116 |
+
face_align = face_align if face_align is not None else True
|
| 117 |
background_enhance = background_enhance if background_enhance is not None else True
|
| 118 |
face_upsample = face_upsample if face_upsample is not None else True
|
| 119 |
upscale = upscale if (upscale is not None and upscale > 0) else 2
|
| 120 |
+
|
| 121 |
+
has_aligned = not face_align
|
| 122 |
upscale = 1 if has_aligned else upscale
|
| 123 |
|
| 124 |
img = cv2.imread(str(image), cv2.IMREAD_COLOR)
|