.

modules/toXML.py

@@ -86,6 +86,8 @@ 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