Ali Mohsin commited on
Commit
d207705
·
1 Parent(s): 224f0c5

feat: Conditionally lower detection threshold and add debug logging for threshold changes and input image statistics.

Browse files
Files changed (2) hide show
  1. SMPL/smpl_regressor.py +8 -0
  2. app.py +4 -1
SMPL/smpl_regressor.py CHANGED
@@ -60,6 +60,14 @@ class SMPL_Regressor:
60
  def create_bev_model(self,fix_body=False):
61
  settings = bev.main.default_settings
62
  settings.mode = 'video'
 
 
 
 
 
 
 
 
63
  bev_model = MyBEV(settings,fix_body=fix_body)
64
  return bev_model
65
 
 
60
  def create_bev_model(self,fix_body=False):
61
  settings = bev.main.default_settings
62
  settings.mode = 'video'
63
+
64
+ # Attempt to lower detection threshold
65
+ if hasattr(settings, 'center_thresh'):
66
+ print(f"DEBUG: Lowering center_thresh from {settings.center_thresh} to 0.02", flush=True)
67
+ settings.center_thresh = 0.02
68
+ else:
69
+ print("DEBUG: settings has no center_thresh. Available:", dir(settings), flush=True)
70
+
71
  bev_model = MyBEV(settings,fix_body=fix_body)
72
  return bev_model
73
 
app.py CHANGED
@@ -238,7 +238,10 @@ def process_frame(image, garment_name, enable_tryon):
238
  print("Error: Frame processor is None inside process_frame")
239
  return image
240
 
241
- # print("Running inference...", flush=True) # Uncomment for spammy logs
 
 
 
242
  output_bgr = frame_processor(img_bgr)
243
 
244
  if output_bgr is None:
 
238
  print("Error: Frame processor is None inside process_frame")
239
  return image
240
 
241
+ # Debug Image Stats
242
+ h, w, c = img_bgr.shape
243
+ print(f"Inference Input - Shape: {h}x{w}, Mean: {img_bgr.mean():.2f}, Max: {img_bgr.max()}", flush=True)
244
+
245
  output_bgr = frame_processor(img_bgr)
246
 
247
  if output_bgr is None: