dreamlessx commited on
Commit
fedb187
·
1 Parent(s): ffb9ba7

Fix distortion: return original image instead of stretched hstack

Browse files
Files changed (1) hide show
  1. app.py +2 -4
app.py CHANGED
@@ -239,9 +239,6 @@ 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
- # 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,7 +254,8 @@ def process_image(image_rgb, procedure, intensity):
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())
 
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
  f"Processing time: {elapsed:.2f}s\n"
255
  f"Mode: TPS (CPU)"
256
  )
257
+ # Return original as 4th output instead of stretched side-by-side
258
+ return wireframe_rgb, mask_vis, composited_rgb, image_rgb_512, info
259
 
260
  except Exception as exc:
261
  logger.error("Processing failed: %s\n%s", exc, traceback.format_exc())