dont create pool with message flow

modules/eval.py

@@ -240,7 +240,6 @@ def regroup_elements_by_pool(boxes, labels, scores, keypoints, class_dict, iou_t
 
         if elements_not_in_pool:
             elements_not_in_pool_to_delete = []
-            new_pool_index = len(labels)
             labels = np.append(labels, list(class_dict.values()).index('pool'))
             #find the messageflow,pool and lane in the elements_not_in_pool
             for i in elements_not_in_pool:
@@ -248,7 +247,9 @@ def regroup_elements_by_pool(boxes, labels, scores, keypoints, class_dict, iou_t
                     elements_not_in_pool_to_delete.append(i)
             #delete the messageflow from the elements_not_in_pool
             elements_not_in_pool = [i for i in elements_not_in_pool if i not in elements_not_in_pool_to_delete]
-            pool_dict[new_pool_index] = elements_not_in_pool
+            if len(elements_not_in_pool) > 0:
+                new_pool_index = len(labels)
+                pool_dict[new_pool_index] = elements_not_in_pool
 
     # Separate empty pools
     non_empty_pools = {k: v for k, v in pool_dict.items() if v}

modules/toXML.py

@@ -86,8 +86,6 @@ def align_boxes(pred, size):
         else:
             if pool_index >= len(modified_pred['boxes']):
                 print("Problem with the index")
-                #delete the pool
-                modified_pred['pool_dict'].pop(pool_index)
                 continue
             min_x, min_y, max_x, max_y = modified_pred['boxes'][pool_index]
         pool_width = max_x - min_x  
@@ -97,7 +95,7 @@ def align_boxes(pred, size):
             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])
-            modified_pred['boxes'] = np.append(modified_pred['boxes'], [new_box], axis=0)
+            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]