Spaces:
Running
Running
test print for fixing bug
Browse files- modules/toXML.py +2 -0
modules/toXML.py
CHANGED
@@ -165,6 +165,8 @@ def create_XML(full_pred, text_mapping, size_scale, scale):
|
|
165 |
|
166 |
# Calculate the bounding box for the pool
|
167 |
#if len(keep_elements) == 0:
|
|
|
|
|
168 |
min_x, min_y, max_x, max_y = full_pred['boxes'][pool_index]
|
169 |
pool_width = max_x - min_x
|
170 |
pool_height = max_y - min_y
|
|
|
165 |
|
166 |
# Calculate the bounding box for the pool
|
167 |
#if len(keep_elements) == 0:
|
168 |
+
print(full_pred['pool_dict'])
|
169 |
+
print(full_pred['boxes'])
|
170 |
min_x, min_y, max_x, max_y = full_pred['boxes'][pool_index]
|
171 |
pool_width = max_x - min_x
|
172 |
pool_height = max_y - min_y
|