Spaces:
Running
Running
correct bug, scale
Browse files- 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
|
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
|