Spaces:
Sleeping
Sleeping
Update utils.py
Browse files
utils.py
CHANGED
@@ -334,7 +334,7 @@ def remove_overlap_new(boxes, iou_threshold, ocr_bbox=None):
|
|
334 |
'content': None
|
335 |
})
|
336 |
else:
|
337 |
-
filtered_boxes.append(
|
338 |
return filtered_boxes # Optionally, you could return torch.tensor(filtered_boxes) if needed
|
339 |
|
340 |
|
|
|
334 |
'content': None
|
335 |
})
|
336 |
else:
|
337 |
+
filtered_boxes.append(box1_elem)
|
338 |
return filtered_boxes # Optionally, you could return torch.tensor(filtered_boxes) if needed
|
339 |
|
340 |
|