BenjiELCA commited on
Commit
99ad6ae
·
1 Parent(s): ca37b38

add margin

Browse files
Files changed (1) hide show
  1. 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//20, min_y - marge, min_x + pool_width + marge//20, min_y + pool_height + 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):