correct overlap pool

modules/eval.py

@@ -372,7 +372,9 @@ def last_correction(boxes, labels, scores, keypoints, links, best_points, pool_d
 
     #concatenate the delete_elements and the delete_pool
     delete_elements = delete_elements + delete_pool
-
+    #delete double value in delete_elements
+    delete_elements = list(set(delete_elements))
+    
     boxes = np.delete(boxes, delete_elements, axis=0)
     labels = np.delete(labels, delete_elements)
     scores = np.delete(scores, delete_elements)

modules/toXML.py

@@ -83,25 +83,32 @@ def align_boxes(pred, size):
         size_elements = get_size_elements(st.session_state.size_scale)
         if len(keep_elements) != 0:
             min_x, min_y, max_x, max_y = calculate_pool_bounds(modified_pred['boxes'],modified_pred['labels'], keep_elements, size_elements)
+            marge = 50
         else:
             if pool_index >= len(modified_pred['boxes']):
                 print("Problem with the index")
                 continue
             min_x, min_y, max_x, max_y = modified_pred['boxes'][pool_index]
+            marge = 0
+
         pool_width = max_x - min_x  
         pool_height = max_y - min_y  
         if pool_width < 300 or pool_height < 30:
             error("The pool is maybe too small, please add more elements or increase the scale by zooming on the image.")
             continue
+
         if pool_index >= len(modified_pred['boxes']):
-            new_box = np.array([min_x - 50, min_y - 50, min_x + pool_width + 50, min_y + pool_height + 50])
+            new_box = np.array([min_x - marge, min_y - marge//2, min_x + pool_width + marge, min_y + pool_height + marge//2])
             modified_pred['boxes'] = np.append(modified_pred['boxes'], [new_box], axis=0)
         else:
-            modified_pred['boxes'][pool_index] = [min_x -50, min_y-50, min_x+pool_width+50, min_y+pool_height+50]
+            modified_pred['boxes'][pool_index] = [min_x -marge, min_y-marge//2, min_x+pool_width+marge, min_y+pool_height+marge//2]
 
     min_left,max_right = 0, 0
     if len(pred['pool_dict'])>1:
         for pool_index, element_indices in pred['pool_dict'].items():
+                if pool_index >= len(modified_pred['boxes']):
+                    print(f"Problem with the index {pool_index} with a length of {len(modified_pred['boxes'])}")
+                    continue
                 x1, y1, x2, y2 = modified_pred['boxes'][pool_index]
                 left = x1
                 right = x2
@@ -111,6 +118,9 @@ def align_boxes(pred, size):
                     max_right = right
 
         for pool_index, element_indices in pred['pool_dict'].items():
+                if pool_index >= len(modified_pred['boxes']):
+                    #print(f"Problem with the index {pool_index} with a length of {len(modified_pred['boxes'])}")
+                    continue
                 x1, y1, x2, y2 = modified_pred['boxes'][pool_index]
                 if x1 > min_left:
                     x1 = min_left