--- a +++ b/monai 0.5.0/deprecated/init.py @@ -0,0 +1,59 @@ +import argparse +import os +import SimpleITK as sitk + +image = sitk.ReadImage('Data_folder/images/train/image0.nii') +image_spacing = image.GetSpacing() + +class Options(): + + """This class defines options used during both training and test time.""" + + def __init__(self): + """Reset the class; indicates the class hasn't been initailized""" + self.initialized = False + + def initialize(self, parser): + + # basic parameters + parser.add_argument('--images_folder', type=str, default='./Data_folder/images') + parser.add_argument('--labels_folder', type=str, default='./Data_folder/labels') + parser.add_argument('--increase_factor_data', default=4, help='Increase data number per epoch') + parser.add_argument('--preload', type=str, default=None) + parser.add_argument('--gpu_ids', type=str, default='2,3', help='gpu ids: e.g. 0 0,1,2, 0,2. use -1 for CPU') + parser.add_argument('--workers', default=8, type=int, help='number of data loading workers') + + # dataset parameters + parser.add_argument('--patch_size', default=(128, 128, 64), help='Size of the patches extracted from the image') + parser.add_argument('--spacing', default=image_spacing, help='Original Resolution') + parser.add_argument('--resolution', default=None, help='New Resolution, if you want to resample the data') + parser.add_argument('--batch_size', type=int, default=6, help='batch size') + parser.add_argument('--in_channels', default=1, type=int, help='Channels of the input') + parser.add_argument('--out_channels', default=1, type=int, help='Channels of the output') + + # training parameters + parser.add_argument('--epochs', default=200, help='Number of epochs') + parser.add_argument('--lr', default=0.001, help='Learning rate') + + # Inference + # This is just a trick to make the predict script working + parser.add_argument('--result', default=None, help='Keep this empty and go to predict_single_image script') + parser.add_argument('--weights', default=None, help='Keep this empty and go to predict_single_image script') + + self.initialized = True + return parser + + def parse(self): + if not self.initialized: + parser = argparse.ArgumentParser(formatter_class=argparse.ArgumentDefaultsHelpFormatter) + parser = self.initialize(parser) + opt = parser.parse_args() + # set gpu ids + if opt.gpu_ids != '-1': + os.environ["CUDA_VISIBLE_DEVICES"] = opt.gpu_ids + return opt + + + + +