update transfomer and config

Author: taigw

examples/prostate/config/train_test.cfg

@@ -85,9 +85,9 @@ mini_batch_size  = 1
 mini_patch_shape = None
 mini_patch_stride= None
 
-checkpoint_name  = model/unet3d/model_12000.pt
+checkpoint_name  = model/unet3d/model_15000.pt
 save_probability = False 
-output_dir       = result_12k
+output_dir       = result
 evaluation_mode  = True
 test_time_dropout = False
 

pymic/io/transform3d.py

@@ -363,7 +363,8 @@ class RandomCrop(object):
 
     def __init__(self, output_size, fg_focus = False, fg_ratio = 0.0, mask_label = None,  inverse = True):
         assert isinstance(output_size, (list, tuple))
-        assert isinstance(mask_label, (list, tuple))
+        if(mask_label is not None):
+            assert isinstance(mask_label, (list, tuple))
         self.output_size = output_size
         self.inverse  = inverse
         self.fg_focus = fg_focus