Spaces:
Running
Running
add margin
Browse files- modules/toXML.py +1 -1
modules/toXML.py
CHANGED
@@ -114,7 +114,7 @@ def expand_pool_bounding_boxes(modified_pred, pred, size_elements):
|
|
114 |
error("The pool is maybe too small, please add more elements or increase the scale by zooming on the image.")
|
115 |
continue
|
116 |
|
117 |
-
modified_pred['boxes'][position] = [min_x - marge
|
118 |
|
119 |
# Adjust left and right boundaries of all pools
|
120 |
def adjust_pool_boundaries(modified_pred, pred):
|
|
|
114 |
error("The pool is maybe too small, please add more elements or increase the scale by zooming on the image.")
|
115 |
continue
|
116 |
|
117 |
+
modified_pred['boxes'][position] = [min_x - marge, min_y - marge//10, min_x + pool_width + marge, min_y + pool_height + marge//10]
|
118 |
|
119 |
# Adjust left and right boundaries of all pools
|
120 |
def adjust_pool_boundaries(modified_pred, pred):
|