예제 #1
0
 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)
예제 #2
0
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]