Update app.py

app.py

@@ -225,13 +225,22 @@ async def detect_multiple_dogs(image, conf_threshold=0.35, iou_threshold=0.5):
 def filter_detections(dogs, image_size):
     filtered_dogs = []
     image_area = image_size[0] * image_size[1]
-    
+    num_dogs = len(dogs)
+
+    # 根據檢測到的狗的數量動態調整閾值
+    if num_dogs > 5:
+        min_ratio, max_ratio = 0.003, 0.5
+    elif num_dogs > 2:
+        min_ratio, max_ratio = 0.005, 0.6
+    else:
+        min_ratio, max_ratio = 0.01, 0.7
+
     for dog in dogs:
-        _, _, box = dog
+        _, confidence, box = dog
         dog_area = (box[2] - box[0]) * (box[3] - box[1])
         area_ratio = dog_area / image_area
         
-        if 0.01 < area_ratio < 0.9:  # 過濾掉太小或太大的檢測框
+        if min_ratio < area_ratio < max_ratio:
             filtered_dogs.append(dog)
     
     return filtered_dogs