change margin with the size of a task

modules/toXML.py

@@ -94,10 +94,10 @@ def align_x_coordinates(modified_pred, x_groups, class_dict, size):
                 ]
 
 # Expand the pool bounding boxes to fit the aligned elements
-def expand_pool_bounding_boxes(modified_pred, pred, size_elements):
+def expand_pool_bounding_boxes(modified_pred, size_elements):
     for idx, (pool_index, keep_elements) in enumerate(modified_pred['pool_dict'].items()):
         if len(keep_elements) != 0:
-            marge = 50
+            marge = size_elements['task'][1] // 2
         else:
             marge = 0
 
@@ -114,7 +114,7 @@ def expand_pool_bounding_boxes(modified_pred, pred, size_elements):
             error("The pool is maybe too small, please add more elements or increase the scale by zooming on the image.")
             continue
         
-        modified_pred['boxes'][position] = [min_x - marge, min_y - marge//10, min_x + pool_width + marge, min_y + pool_height + marge//10]
+        modified_pred['boxes'][position] = [min_x - marge, min_y - marge//2, min_x + pool_width + marge, min_y + pool_height + marge//2]
 
 # Adjust left and right boundaries of all pools
 def adjust_pool_boundaries(modified_pred, pred):
@@ -149,7 +149,7 @@ def align_boxes(pred, size, class_dict):
     align_elements_within_pool(modified_pred, pool_groups, class_dict, size)
     
     if len(pred['pool_dict']) > 1:
-        expand_pool_bounding_boxes(modified_pred, pred, size)
+        expand_pool_bounding_boxes(modified_pred, size)
         adjust_pool_boundaries(modified_pred, pred)
 
     return modified_pred['boxes']