def test_bbox_vflip():
assert F.bbox_vflip((0.1, 0.2, 0.6, 0.5), 100, 200) == (0.1, 0.5, 0.6, 0.8)
def test_bbox_hflip():
assert F.bbox_hflip((0.1, 0.2, 0.6, 0.5), 100, 200) == (0.4, 0.2, 0.9, 0.5)
@pytest.mark.parametrize(
["code", "func"],
[
[0, F.bbox_vflip],
[1, F.bbox_hflip],
[
-1, lambda bbox, rows, cols: F.bbox_vflip(
F.bbox_hflip(bbox, rows, cols), rows, cols)
],
],
)
def test_bbox_flip(code, func):
rows, cols = 100, 200
bbox = [0.1, 0.2, 0.6, 0.5]
assert F.bbox_flip(bbox, code, rows, cols) == func(bbox, rows, cols)
def test_crop_bbox_by_coords():
cropped_bbox = A.crop_bbox_by_coords((0.5, 0.2, 0.9, 0.7),
(18, 18, 82, 82), 64, 64, 100, 100)
assert cropped_bbox == (0.5, 0.03125, 1.125, 0.8125)
def test_bbox_hflip():
assert F.bbox_hflip((0.1, 0.2, 0.6, 0.5), 100, 200) == (0.4, 0.2, 0.9, 0.5)
def test_bbox_hflip(bbox, expected_output, input_type):
bbox = input_type(bbox)
flipped_bbox = F.bbox_hflip(bbox, cols=200, rows=100)
assert isinstance(flipped_bbox, tuple)
assert np.array_equal(flipped_bbox, expected_output)
def apply_to_bbox(self, bbox, **params):
return af.bbox_hflip(bbox, **params)
def test_bbox_hflip():
assert F.bbox_hflip([0.1, 0.2, 0.6, 0.5], 100, 200) == [0.4, 0.2, 0.9, 0.5]
assert width == 2
assert np.array_equal(resized_img, expected)
def test_bbox_vflip():
assert F.bbox_vflip([0.1, 0.2, 0.6, 0.5], 100, 200) == [0.1, 0.5, 0.6, 0.8]
def test_bbox_hflip():
assert F.bbox_hflip([0.1, 0.2, 0.6, 0.5], 100, 200) == [0.4, 0.2, 0.9, 0.5]
@pytest.mark.parametrize(['code', 'func'], [
[0, F.bbox_vflip],
[1, F.bbox_hflip],
[-1, lambda bbox, rows, cols: F.bbox_vflip(F.bbox_hflip(bbox, rows, cols), rows, cols)],
])
def test_bbox_flip(code, func):
rows, cols = 100, 200
bbox = [0.1, 0.2, 0.6, 0.5]
assert F.bbox_flip(bbox, code, rows, cols) == func(bbox, rows, cols)
def test_crop_bbox_by_coords():
cropped_bbox = F.crop_bbox_by_coords([0.5, 0.2, 0.9, 0.7], (18, 18, 82, 82), 64, 64, 100, 100)
assert cropped_bbox == [0.5, 0.03125, 1.125, 0.8125]
def test_bbox_center_crop():
cropped_bbox = F.bbox_center_crop([0.5, 0.2, 0.9, 0.7], 64, 64, 100, 100)
assert cropped_bbox == [0.5, 0.03125, 1.125, 0.8125]
def test_bbox_hflip(bbox, expected_output, input_type):
bbox = input_type(bbox)
flipped_bbox = F.bbox_hflip(bbox, cols=200, rows=100)
assert isinstance(flipped_bbox, tuple)
assert np.array_equal(flipped_bbox, expected_output)
