30
30
31
31
from monai .config .deviceconfig import USE_COMPILED
32
32
from monai .data import create_test_image_2d , create_test_image_3d
33
- from monai .transforms import Randomizable
34
33
from monai .utils import ensure_tuple , optional_import , set_determinism
35
34
from monai .utils .module import get_torch_version_tuple
36
35
@@ -435,25 +434,14 @@ def _call_original_func(name, module, *args, **kwargs):
435
434
return f (* args , ** kwargs )
436
435
437
436
438
- class NumpyImageTestCase2D (unittest .TestCase , Randomizable ):
437
+ class NumpyImageTestCase2D (unittest .TestCase ):
439
438
im_shape = (128 , 64 )
440
439
input_channels = 1
441
440
output_channels = 4
442
441
num_classes = 3
443
442
444
- def randomize (self , data = None ):
445
- return create_test_image_2d (
446
- width = self .im_shape [0 ],
447
- height = self .im_shape [1 ],
448
- num_objs = 4 ,
449
- rad_max = 20 ,
450
- noise_max = 0 ,
451
- num_seg_classes = self .num_classes ,
452
- random_state = self .R ,
453
- )
454
-
455
443
def setUp (self ):
456
- im , msk = self .randomize ( )
444
+ im , msk = create_test_image_2d ( self .im_shape [ 0 ], self . im_shape [ 1 ], 4 , 20 , 0 , self . num_classes )
457
445
458
446
self .imt = im [None , None ]
459
447
self .seg1 = (msk [None , None ] > 0 ).astype (np .float32 )
@@ -468,26 +456,14 @@ def setUp(self):
468
456
self .segn = torch .tensor (self .segn )
469
457
470
458
471
- class NumpyImageTestCase3D (unittest .TestCase , Randomizable ):
459
+ class NumpyImageTestCase3D (unittest .TestCase ):
472
460
im_shape = (64 , 48 , 80 )
473
461
input_channels = 1
474
462
output_channels = 4
475
463
num_classes = 3
476
464
477
- def randomize (self , data = None ):
478
- return create_test_image_3d (
479
- height = self .im_shape [0 ],
480
- width = self .im_shape [1 ],
481
- depth = self .im_shape [2 ],
482
- num_objs = 4 ,
483
- rad_max = 20 ,
484
- noise_max = 0 ,
485
- num_seg_classes = self .num_classes ,
486
- random_state = self .R ,
487
- )
488
-
489
465
def setUp (self ):
490
- im , msk = self .randomize ( )
466
+ im , msk = create_test_image_3d ( self .im_shape [ 0 ], self . im_shape [ 1 ], self . im_shape [ 2 ], 4 , 20 , 0 , self . num_classes )
491
467
492
468
self .imt = im [None , None ]
493
469
self .seg1 = (msk [None , None ] > 0 ).astype (np .float32 )
0 commit comments