--- a +++ b/inst/deepbleed/tools/parse.py @@ -0,0 +1,41 @@ +# @author: msharrock +# version: 0.0.1 + +''' +Command Line Argument Parsing Tools + +''' + +import os +import argparse + +def args(mode): + + ''' + Command Line Argument Interpreter + + Params: + - session: string, 'predict','train','validate' + + Returns: + - args: object, accessible representation of arguments + ''' + if mode == 'predict': + + parser = argparse.ArgumentParser(description="Arguments for Prediction of Hemorrhage") + + parser.add_argument('-i','--indir', required=True, action='store', + dest='IN_DIR', help='input directory') + parser.add_argument('-o','--outdir', required=True, action='store', + dest='OUT_DIR', help='output directory') + parser.add_argument('-w','--weights', required=True, action='store', + dest='weights', help='tf model weights') + parser.add_argument('--gpus', required=False, action='store', + type=int, dest='GPUS', + help='number of GPUs to utilize') + parser.add_argument('--cpus', required=False, action='store', + type=int, dest='CPUS', help='number of CPU cores') + parser.add_argument('-v', '--verbose', required=False, action='store_true', + help='verbose and timed script') + + return parser.parse_args() \ No newline at end of file