def test_deterministic_one_or_other():
aug = ReplayCompose([OneOrOther(HorizontalFlip(), Blur())], p=1)
for i in range(10):
image = (np.random.random((8, 8)) * 255).astype(np.uint8)
image2 = np.copy(image)
data = aug(image=image)
assert "replay" in data
data2 = ReplayCompose.replay(data["replay"], image=image2)
assert np.array_equal(data["image"], data2["image"])
def test_crop_boxes_replay_compose():
image = np.ones((512, 384, 3))
bboxes = [(78, 42, 142, 80), (32, 12, 42, 72), (200, 100, 300, 200)]
labels = [0, 1, 2]
transform = ReplayCompose(
[RandomCrop(256, 256, p=1.0)],
bbox_params=BboxParams(format="pascal_voc", min_area=16, label_fields=["labels"]),
)
input_data = dict(image=image, bboxes=bboxes, labels=labels)
transformed = transform(**input_data)
transformed2 = ReplayCompose.replay(transformed["replay"], **input_data)
np.testing.assert_almost_equal(transformed["bboxes"], transformed2["bboxes"])