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- SMPL/smpl_regressor.py +8 -0
- 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 |
-
#
|
|
|
|
|
|
|
|
|
|
| 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:
|