Update app.py

Filter filter code update

app.py

@@ -92,7 +92,7 @@ def filter_bbox_outliers(detections: List[Dict],
     
     Args:
         detections: List of detection dictionaries with 'bbox', 'label', 'score'
-        method: 'iqr' (Interquartile Range) 
+        method: 'iqr' (Interquartile Range) or 'zscore' (Z-score) or 'percentile'
         threshold: Multiplier for IQR method or Z-score threshold
         min_score: Minimum confidence score to keep detection
         
@@ -109,8 +109,20 @@ def filter_bbox_outliers(detections: List[Dict],
     areas = [calculate_bbox_area(det['bbox']) for det in detections]
     areas = np.array(areas)
     
+    if method == 'iqr':
+        # IQR method
+        q1 = np.percentile(areas, 25)
+        q3 = np.percentile(areas, 75)
+        iqr = q3 - q1
         
-    if method == 'zscore':
+        # Define outlier boundaries
+        lower_bound = q1 - threshold * iqr
+        upper_bound = q3 + threshold * iqr
+        
+        # Find valid indices (non-outliers)
+        valid_indices = np.where((areas >= lower_bound) & (areas <= upper_bound))[0]
+        
+    elif method == 'zscore':
         # Z-score method
         mean_area = np.mean(areas)
         std_area = np.std(areas)
@@ -118,9 +130,8 @@ def filter_bbox_outliers(detections: List[Dict],
         z_scores = np.abs((areas - mean_area) / std_area)
         valid_indices = np.where(z_scores <= threshold)[0]
         
-    
     else:
-        raise ValueError("Method must be 'iqr', 'zscore', or 'percentile'")
+        raise ValueError("Method must be 'iqr' or 'zscore'")
     
     # Return filtered detections
     filtered_detections = [detections[i] for i in valid_indices]
@@ -167,7 +178,7 @@ def detect_objects_owlv2(text_query, image, threshold=0.1):
                 detections.append(detection)
 
         print(detections)
-        return filter_bbox_outliers(detections,method = 'iqr'),image
+        return filter_bbox_outliers(detections,method = 'zscore'),image
         
     except Exception as e:
         print(f"Detection error: {e}")