Update app.py

app.py

@@ -107,9 +107,10 @@ def process_image(job_id, image_path, object_type, multiplier):
                 upscaled_image = cv2.resize(image, None, fx=scale_factor, fy=scale_factor, interpolation=cv2.INTER_LINEAR)
                 temp_path = "upscaled.jpg"
                 cv2.imwrite(temp_path, upscaled_image)
-                results = box_model.predict(temp_path, confidence=50, overlap=30).json()
+                # Lower confidence threshold to 30 and overlap to 20 for better sensitivity
+                results = box_model.predict(temp_path, confidence=30, overlap=20).json()
             else:
-                results = box_model.predict(image_path, confidence=50, overlap=30).json()
+                results = box_model.predict(image_path, confidence=30, overlap=20).json()
             
             predictions = results.get("predictions", [])
             processed_preds = []
@@ -143,7 +144,7 @@ def process_image(job_id, image_path, object_type, multiplier):
                 except Exception as e:
                     continue
 
-            # Increase the NMS IoU threshold to 0.7 so that boxes that are touching are less likely to be suppressed.
+            # You might experiment with the NMS threshold further if needed.
             box_detections = custom_nms(processed_preds, iou_threshold=0.7)
             jobs[job_id]['progress'] = 60