We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 1482b7b commit 6498a94Copy full SHA for 6498a94
cellseg_models_pytorch/transforms/tests/test_stronaug.py
@@ -25,19 +25,3 @@ def test_stronaug_transforms(img_sample, op_name):
25
assert tr_img.shape == img_sample.shape
26
assert tr_img.dtype == img_sample.dtype
27
assert tr_img.max() <= 255
28
-
29
30
-def test_strongaug(img_patch_dir, mask_patch_dir):
31
- imp = sorted(img_patch_dir.glob("*"))[0]
32
- mp = sorted(mask_patch_dir.glob("*"))[0]
33
34
- im = FileHandler.read_img(imp)
35
- mask = FileHandler.read_mat(mp)["inst_map"]
36
- mask2 = FileHandler.read_mat(mp)["type_map"]
37
38
- sa = AlbuStrongAugment()
39
- tr_data = sa(image=im, masks=[mask, mask2])
40
41
- assert tr_data["image"].shape == im.shape
42
- assert tr_data["image"].dtype == im.dtype
43
- assert list(tr_data.keys()) == ["image", "masks"]
0 commit comments