Spaces:
Sleeping
Sleeping
dreamlessx commited on
Commit ·
ffb9ba7
1
Parent(s): dc32c69
Fix side-by-side before/after output in Single Procedure tab
Browse files
app.py
CHANGED
|
@@ -239,6 +239,9 @@ def process_image(image_rgb, procedure, intensity):
|
|
| 239 |
composited = mask_composite(warped, image_bgr, mask)
|
| 240 |
composited_rgb = cv2.cvtColor(composited, cv2.COLOR_BGR2RGB)
|
| 241 |
|
|
|
|
|
|
|
|
|
|
| 242 |
displacement = np.mean(
|
| 243 |
np.linalg.norm(manipulated.pixel_coords - face.pixel_coords, axis=1)
|
| 244 |
)
|
|
@@ -254,7 +257,7 @@ def process_image(image_rgb, procedure, intensity):
|
|
| 254 |
f"Processing time: {elapsed:.2f}s\n"
|
| 255 |
f"Mode: TPS (CPU)"
|
| 256 |
)
|
| 257 |
-
return wireframe_rgb, mask_vis, composited_rgb,
|
| 258 |
|
| 259 |
except Exception as exc:
|
| 260 |
logger.error("Processing failed: %s\n%s", exc, traceback.format_exc())
|
|
|
|
| 239 |
composited = mask_composite(warped, image_bgr, mask)
|
| 240 |
composited_rgb = cv2.cvtColor(composited, cv2.COLOR_BGR2RGB)
|
| 241 |
|
| 242 |
+
# Side-by-side comparison: original on left, result on right
|
| 243 |
+
sbs = np.hstack([image_rgb_512, composited_rgb])
|
| 244 |
+
|
| 245 |
displacement = np.mean(
|
| 246 |
np.linalg.norm(manipulated.pixel_coords - face.pixel_coords, axis=1)
|
| 247 |
)
|
|
|
|
| 257 |
f"Processing time: {elapsed:.2f}s\n"
|
| 258 |
f"Mode: TPS (CPU)"
|
| 259 |
)
|
| 260 |
+
return wireframe_rgb, mask_vis, composited_rgb, sbs, info
|
| 261 |
|
| 262 |
except Exception as exc:
|
| 263 |
logger.error("Processing failed: %s\n%s", exc, traceback.format_exc())
|