Update app.py

app.py

@@ -6,6 +6,30 @@ import cv2
 
 model = YOLO("yolo26s.pt")
 
+def is_valid_person(x1, y1, x2, y2, img_h, img_w):
+    """Filter out partial detections — shoes, legs, hands etc.
+    A real full/half person must:
+    - Be tall enough relative to image height
+    - Have a portrait-ish aspect ratio (taller than wide)
+    - Not be tiny
+    """
+    box_w = x2 - x1
+    box_h = y2 - y1
+
+    # Minimum size — must be at least 4% of image height
+    min_height = img_h * 0.04
+    if box_h < min_height:
+        return False
+
+    # Aspect ratio — person box should be taller than wide (portrait)
+    # Allow up to 1.2 wide:tall ratio to catch slightly leaning people
+    aspect_ratio = box_w / box_h
+    if aspect_ratio > 1.2:
+        return False
+
+    return True
+
+
 def run_tiled_detection(img_bgr, tile_size=640, overlap=0.2, conf=0.50):
     h, w = img_bgr.shape[:2]
     step = int(tile_size * (1 - overlap))
@@ -19,11 +43,13 @@ def run_tiled_detection(img_bgr, tile_size=640, overlap=0.2, conf=0.50):
             results = model(tile, classes=[0], conf=conf, verbose=False)
             for box in results[0].boxes:
                 bx1, by1, bx2, by2 = map(int, box.xyxy[0])
-                all_boxes.append({
-                    "x1": bx1 + x, "y1": by1 + y,
-                    "x2": bx2 + x, "y2": by2 + y,
-                    "conf": float(box.conf[0])
-                })
+                # Translate to full image coords
+                fx1, fy1, fx2, fy2 = bx1+x, by1+y, bx2+x, by2+y
+                if is_valid_person(fx1, fy1, fx2, fy2, h, w):
+                    all_boxes.append({
+                        "x1": fx1, "y1": fy1, "x2": fx2, "y2": fy2,
+                        "conf": float(box.conf[0])
+                    })
 
     if not all_boxes:
         return []
@@ -37,13 +63,14 @@ def run_tiled_detection(img_bgr, tile_size=640, overlap=0.2, conf=0.50):
 
 
 def run_standard_detection(img_bgr, conf=0.50):
+    h, w = img_bgr.shape[:2]
     results = model(img_bgr, classes=[0], conf=conf, verbose=False)
-    return [
-        {"x1": int(b.xyxy[0][0]), "y1": int(b.xyxy[0][1]),
-         "x2": int(b.xyxy[0][2]), "y2": int(b.xyxy[0][3]),
-         "conf": float(b.conf[0])}
-        for b in results[0].boxes
-    ]
+    detections = []
+    for b in results[0].boxes:
+        x1, y1, x2, y2 = int(b.xyxy[0][0]), int(b.xyxy[0][1]), int(b.xyxy[0][2]), int(b.xyxy[0][3])
+        if is_valid_person(x1, y1, x2, y2, h, w):
+            detections.append({"x1": x1, "y1": y1, "x2": x2, "y2": y2, "conf": float(b.conf[0])})
+    return detections
 
 
 def count_people(image, mode):
@@ -120,7 +147,7 @@ with gr.Blocks(title="People Counter — YOLO26") as demo:
 
     gr.HTML("""
         <div style='text-align:center; color:#888; font-size:0.8rem; padding: 1.5rem 0 0.5rem;'>
-            Powered by <strong>Ultralytics YOLO26</strong> · Model: yolo26s · Conf: 0.50
+            Powered by <strong>Ultralytics YOLO26</strong> · Model: yolo26s · Conf: 0.50 · Full-body filter ON
         </div>
     """)