Update app.py

app.py

@@ -144,21 +144,27 @@ def process_image(job_id, image_path, object_type, multiplier):
                 except Exception as e:
                     continue
 
-            # You might experiment with the NMS threshold further if needed.
+            # Increase the NMS IoU threshold to 0.7 so that touching boxes are less likely to be suppressed.
             box_detections = custom_nms(processed_preds, iou_threshold=0.7)
             jobs[job_id]['progress'] = 60
 
             # (Optional) Detect ArUco marker for measurement.
-            marker_real_width_cm = 10.0  
+            # Set the real-world width of your marker in centimeters.
+            # For a marker that is 5cm x 5cm, use 5.0.
+            marker_real_width_cm = 5.0  
             try:
                 gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
+                # Use a predefined dictionary. You can try DICT_6X6_250 (or DICT_4X4_50 if you prefer).
                 aruco_dict = cv2.aruco.getPredefinedDictionary(cv2.aruco.DICT_6X6_250)
+                # Alternatively:
+                # aruco_dict = cv2.aruco.getPredefinedDictionary(cv2.aruco.DICT_4X4_50)
                 if hasattr(cv2.aruco, 'DetectorParameters_create'):
                     aruco_params = cv2.aruco.DetectorParameters_create()
                 else:
                     aruco_params = cv2.aruco.DetectorParameters()
                 corners, ids, _ = cv2.aruco.detectMarkers(gray, aruco_dict, parameters=aruco_params)
                 if ids is not None and len(corners) > 0:
+                    # Use the first detected marker as reference.
                     marker_corners = corners[0].reshape((4, 2))
                     cv2.aruco.drawDetectedMarkers(image, corners, ids)
                     side_lengths = []