i was creating a big pool twice

modules/eval.py

@@ -232,21 +232,11 @@ def regroup_elements_by_pool(boxes, labels, scores, keypoints, class_dict, iou_t
                 keypoints = np.append(keypoints, np.zeros((1, 2, 3)), axis=0)
                 pool_dict[new_pool_index] = new_elements_not_in_pool
                 print(f"Created a new pool index {new_pool_index} with elements: {new_elements_not_in_pool}")
-    else:
-        all_elements = [i for i in range(len(boxes))]
-        new_pool_index = len(labels)
-        box = calculate_pool_bounds(boxes, labels, all_elements, None)
-        boxes = np.append(boxes, [box], axis=0)
-        labels = np.append(labels, list(class_dict.values()).index('pool'))
-        scores = np.append(scores, 1.0)
-        keypoints = np.append(keypoints, np.zeros((1, 2, 3)), axis=0)
-        pool_dict[new_pool_index] = all_elements
-        print(f"Created a super pool {new_pool_index} with elements: {all_elements}")
        
     non_empty_pools = {k: v for k, v in pool_dict.items() if v}
     empty_pools = {k: v for k, v in pool_dict.items() if not v}
     pool_dict = {**non_empty_pools, **empty_pools}
-    
+
     return pool_dict, boxes, labels, scores, keypoints
 
 

modules/toXML.py

@@ -236,7 +236,7 @@ def create_XML(full_pred, text_mapping, size_scale, scale):
 
 def create_big_pool(full_pred, text_mapping):
     # If no pools or lanes are detected, create a single pool with all elements
-    new_pool_index = len(full_pred['labels'])
+    new_pool_index = 'pool_1'
     size_elements = get_size_elements(st.session_state.size_scale)
     elements_pool = list(range(len(full_pred['boxes'])))
     min_x, min_y, max_x, max_y = calculate_pool_bounds(full_pred['boxes'],full_pred['labels'], elements_pool, size_elements)
@@ -245,7 +245,7 @@ def create_big_pool(full_pred, text_mapping):
     full_pred['pool_dict'][new_pool_index] = elements_pool
     full_pred['BPMN_id'].append('pool_1')
     text_mapping['pool_1'] = 'Process'
-    print(f"Created a new pool index {new_pool_index} with elements: {elements_pool}")
+    print(f"Created a big pool index {new_pool_index} with elements: {elements_pool}")
     return full_pred, text_mapping
 
 def get_size_elements(size_scale):