def apply_to_bbox(self, bbox, x_min=0, x_max=0, y_min=0, y_max=0, **params): h, w = params['rows'], params['cols'] if x_min < 0: x_min = 0 if y_min < 0: y_min = 0 if y_max >= h: y_max = h - 1 if x_max >= w: x_max = w - 1 return F.bbox_crop(bbox, x_min, y_min, x_max, y_max, **params)
def test_bbox_crop(): cropped_bbox = F.bbox_crop([0.5, 0.2, 0.9, 0.7], 24, 24, 64, 64, 100, 100) assert cropped_bbox == [0.65, -0.1, 1.65, 1.15]