Upload app.py

app.py

@@ -137,7 +137,7 @@ CONFIGS = {
         'model_path': 'best_upright.onnx',
         'yaml_path': 'data_upright.yaml',
         'input_size': 640,
-        'confidence_threshold': 0.25,  # Lowered from 0.45 for better detection
+        'confidence_threshold': 0.5,  # 🔧 Match test script confidence (was 0.25)
         'nms_threshold': 0.45
     }
 }
@@ -179,6 +179,7 @@ class ModelManager:
                 
                 # Load model dengan available backend
                 session = None
+                actual_input_size = config['input_size']  # Default fallback
                 
                 if ONNX_AVAILABLE and ort is not None:
                     # Load ONNX session dengan CPU optimization
@@ -194,6 +195,18 @@ class ModelManager:
                         providers=providers,
                         sess_options=session_options
                     )
+                    
+                    # 🔧 AUTO-DETECT input size dari model shape (fix untuk upright model)
+                    try:
+                        input_shape = session.get_inputs()[0].shape
+                        if isinstance(input_shape, (list, tuple)) and len(input_shape) >= 4:
+                            h, w = input_shape[2], input_shape[3]
+                            if isinstance(h, int) and isinstance(w, int) and h > 0 and w > 0:
+                                actual_input_size = h  # gunakan height dari model shape
+                                logger.info(f"🔧 AUTO-DETECTED input size untuk {config_key}: {actual_input_size} (was {config['input_size']})")
+                    except Exception as e:
+                        logger.warning(f"⚠️ Failed to auto-detect input size for {config_key}: {e}")
+                        # Keep using config input_size as fallback
                 else:
                     # For now, only ONNX Runtime is supported for model loading
                     # PyTorch/TensorFlow alternatives would need model conversion
@@ -208,7 +221,7 @@ class ModelManager:
                     'session': session,
                     'class_names': class_names,
                     'input_name': session.get_inputs()[0].name,
-                    'input_size': config['input_size'],
+                    'input_size': actual_input_size,  # 🔧 Gunakan auto-detected input size
                     'confidence': config['confidence_threshold'],
                     'nms': config.get('nms_threshold', 0.45)
                 }