def test_filter_bboxes(bboxes, min_area, min_visibility, target):
filtered_bboxes = filter_bboxes(bboxes,
min_area=min_area,
min_visibility=min_visibility,
rows=100,
cols=100)
assert filtered_bboxes == target
def boxes_postprocessing(self, data):
rows, cols = data['image'].shape[:2]
data['bboxes'] = filter_bboxes(data['bboxes'], rows, cols,
self.min_area, self.min_visibility)
data['bboxes'] = convert_bboxes_from_albumentations(
data['bboxes'], self.bbox_format, rows, cols, check_validity=True)
if self.label_fields:
for idx, field in enumerate(self.label_fields):
field_values = []
for bbox in data['bboxes']:
field_values.append(bbox[4 + idx])
data[field] = field_values
data['bboxes'] = [bbox[:4] for bbox in data['bboxes']]
return data
