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