BenjiELCA commited on
Commit
00888d3
·
1 Parent(s): 1814da2

correct bug, scale

Browse files
Files changed (1) hide show
  1. modules/toXML.py +1 -1
modules/toXML.py CHANGED
@@ -179,7 +179,7 @@ def create_XML(full_pred, text_mapping, size_scale, scale):
179
  size_elements = get_size_elements(size_scale)
180
 
181
  #if there is no pool or lane, create a pool with all elements
182
- if len(full_pred['pool_dict'])==0 or (len(full_pred['pool_dict'])==1 and list(full_pred['pool_dict'])[0]==len(full_pred['labels'])):
183
  full_pred, text_mapping = create_big_pool(full_pred, text_mapping)
184
 
185
  #modify the boxes positions
 
179
  size_elements = get_size_elements(size_scale)
180
 
181
  #if there is no pool or lane, create a pool with all elements
182
+ if len(full_pred['pool_dict'])==0 or (len(full_pred['pool_dict'])==1 and len(full_pred['pool_dict']['pool_1'])==len(full_pred['labels'])):
183
  full_pred, text_mapping = create_big_pool(full_pred, text_mapping)
184
 
185
  #modify the boxes positions