File size: 5,146 Bytes
ab9857f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 |
import sys, getopt
import DeepDeformationMapRegistration.utils.constants as C
import os
def parse_arguments(argv):
try:
opts, args = getopt.getopt(argv, "hg:b:l:r:d:t:i:f:x:p:q:", ["gpu-num=",
"batch-size=",
"loss=",
"remote=",
"debug=",
"debug-training=",
"debug-input-data=",
"destination-folder=",
"destination-folder-fix=",
"training-dataset=",
"test-dataset=",
"help"])
except getopt.GetoptError:
print('\n\t\t--gpu-num:\t\tGPU number to use'
'\n\t\t--batch-size:\t\tsize of the training batch'
'\n\t\t--loss:\t\tLoss function: ncc, mse, dssim'
'\n\t\t--remote:\t\tExecuting the script in The Beast: "True"/"False". Def: False'
'\n\t\t--debug:\t\tEnable debugging logs: "True"/"False". Def: False'
'\n\t\t--debug-training:\t\tEnable debugging training logs: "True"/"False". Def: False'
'\n\t\t--debug-input-data:\t\tEnable debugging input data logs: "True"/"False". Def: False'
'\n\t\t--destination-folder:\t\tName of the folder where to save the generated training files'
'\n\t\t--destination-folder-fixed:\t\tSame as --destination-folder but do not add the timestamp'
'\n\t\t--training-dataset:\t\tPath to the training dataset file'
'\n\t\t--test-dataset:\t\tPath to the test dataset file'
'\n')
sys.exit(2)
for opt, arg in opts:
if opt in ('--help', '-h'):
print('\n\t\t--gpu-num:\t\tGPU number to use\n\t\t--batch-size:\t\tsize of the training batch'
'\n\t\t--loss:\t\tLoss function: ncc, mse, dssim\n')
continue
elif opt in ('--gpu_num', '-g'):
old = C.GPU_NUM
C.GPU_NUM = arg
os.environ['CUDA_VISIBLE_DEVICES'] = C.GPU_NUM
print('\t\tGPU_NUM: {} -> {}'.format(old, C.GPU_NUM))
elif opt in ('--batch-size', '-b'):
old = C.BATCH_SIZE
C.BATCH_SIZE = int(arg)
print('\t\tBATCH_SIZE: {} -> {}'.format(old, C.BATCH_SIZE))
elif opt in ('--destination-folder', '-f'):
old = C.DESTINATION_FOLDER
C.DESTINATION_FOLDER = arg + '_' + C.CUR_DATETIME
print('\t\tDESTINATION_FOLDER: {} -> {}'.format(old, C.DESTINATION_FOLDER))
elif opt in ('--destination-folder-fixed', '-x'):
old = C.DESTINATION_FOLDER
C.DESTINATION_FOLDER = arg
print('\t\tDESTINATION_FOLDER: {} -> {}'.format(old, C.DESTINATION_FOLDER))
elif opt in ('--training-dataset', '-p'):
old = C.TRAINING_DATASET
C.TRAINING_DATASET = arg
print('\t\tTRAINING_DATASET: {} -> {}'.format(old, C.TRAINING_DATASET))
elif opt in ('--test-dataset', '-q'):
old = C.TEST_DATASET
C.TEST_DATASET = arg
print('\t\tTEST_DATASET: {} -> {}'.format(old, C.TEST_DATASET))
elif opt in ('--remote', '-r'):
old = C.REMOTE
if arg.lower() in ('1', 'true', 't'):
C.REMOTE = True
else:
C.REMOTE = False
print('\t\tREMOTE: {} -> {}'.format(old, C.REMOTE))
elif opt in ('--debug', '-d'):
old = C.DEBUG
if arg.lower() in ('1', 'true', 't'):
C.DEBUG = True
else:
C.DEBUG = False
print('\t\tDEBUG: {} -> {}'.format(old, C.DEBUG))
elif opt in ('--debug-training', '-t'):
old = C.DEBUG_TRAINING
if arg.lower() in ('1', 'true', 't'):
C.DEBUG_TRAINING = True
else:
C.DEBUG_TRAINING = False
print('\t\tDEBUG_TRAINING: {} -> {}'.format(old, C.DEBUG_TRAINING))
elif opt in ('--debug-input-data', '-i'):
old = C.DEBUG_INPUT_DATA
if arg.lower() in ('1', 'true', 't'):
C.DEBUG_INPUT_DATA = True
else:
C.DEBUG_INPUT_DATA = False
print('\t\tDEBUG_INPUT_DATA: {} -> {}'.format(old, C.DEBUG_INPUT_DATA))
elif opt in ('--loss', '-l'):
old = C.LOSS_FNC
if arg in ('ncc', 'mse', 'dssim', 'dice'):
C.LOSS_FNC = arg
else:
print('Invalid option for --loss. Expected: "mse", "ncc" or "dssim", got {}'.format(arg))
sys.exit(2)
print('\t\tLOSS_FNC: {} -> {}'.format(old, C.LOSS_FNC))
|