File tree Expand file tree Collapse file tree 1 file changed +0
-16
lines changed
cellseg_models_pytorch/transforms/tests Expand file tree Collapse file tree 1 file changed +0
-16
lines changed Original file line number Diff line number Diff line change @@ -25,19 +25,3 @@ def test_stronaug_transforms(img_sample, op_name):
25
25
assert tr_img .shape == img_sample .shape
26
26
assert tr_img .dtype == img_sample .dtype
27
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" ]
You can’t perform that action at this time.
0 commit comments