| import os, sys |
| currentdir = os.path.dirname(os.path.realpath(__file__)) |
| parentdir = os.path.dirname(currentdir) |
| sys.path.append(parentdir) |
|
|
| from Brain_study.Train_SegmentationGuided import launch_train |
| import argparse |
|
|
| TRAIN_DATASET = '/mnt/EncryptedData1/Users/javier/ext_datasets/IXI_dataset/T1/training' |
|
|
| err = list() |
|
|
| if __name__ == '__main__': |
| parser = argparse.ArgumentParser() |
| parser.add_argument('--dataset', type=str, help='Location of the training data', default=TRAIN_DATASET) |
| parser.add_argument('--validation', type=str, help='Location of the validation data', default=None) |
| parser.add_argument('--similarity', type=str, help='Similarity loss function: mse, ncc, ssim') |
| parser.add_argument('--segmentation', type=str, help='Segmentation loss function: hd, dice') |
| parser.add_argument('--output', type=str, help='Output directory', default=TRAIN_DATASET) |
| parser.add_argument('--gpu', type=str, help='GPU number', default='0') |
| parser.add_argument('--lr', type=float, help='Learning rate', default=1e-4) |
| parser.add_argument('--rw', type=float, help='Regularization weigh', default=2e-2) |
|
|
| args = parser.parse_args() |
|
|
| print('TRAIN ' + args.dataset) |
| launch_train(dataset_folder=args.dataset, |
| validation_folder=args.validation, |
| output_folder=os.path.join(args.output, 'SEGGUIDED_Lsim_{}__Lseg_{}__MET_mse_ncc_ssim'.format(args.similarity, args.segmentation)), |
| gpu_num=args.gpu, |
| lr=args.lr, |
| rw=args.rw, |
| simil=args.similarity, |
| segm=args.segmentation) |
|
|
