fix bug

modules/toXML.py

@@ -84,6 +84,9 @@ def align_boxes(pred, size):
         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)
         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]
         pool_width = max_x - min_x  
         pool_height = max_y - min_y