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):
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" ]
You can’t perform that action at this time.
0 commit comments