.

modules/toXML.py

@@ -168,8 +168,9 @@ def create_XML(full_pred, text_mapping, size_scale, scale):
         
         # Calculate the bounding box for the pool
         #if len(keep_elements) == 0:
-        print(full_pred['pool_dict'])
-        print(full_pred['boxes'])
+        if pool_index >= len(full_pred['boxes']):
+            print("Problem with the index")
+            continue
         min_x, min_y, max_x, max_y = full_pred['boxes'][pool_index]
         pool_width = max_x - min_x
         pool_height = max_y - min_y