Skip to content

Commit 6498a94

Browse files
committed
fix: fix strong aug test
1 parent 1482b7b commit 6498a94

File tree

1 file changed

+0
-16
lines changed

1 file changed

+0
-16
lines changed

cellseg_models_pytorch/transforms/tests/test_stronaug.py

Lines changed: 0 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -25,19 +25,3 @@ def test_stronaug_transforms(img_sample, op_name):
2525
assert tr_img.shape == img_sample.shape
2626
assert tr_img.dtype == img_sample.dtype
2727
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

Comments
 (0)