32 lines
1021 B
Python
32 lines
1021 B
Python
import argparse
|
|
|
|
args = None
|
|
|
|
|
|
def get_args():
|
|
global args
|
|
if args is None:
|
|
args = parse_args()
|
|
return args
|
|
|
|
|
|
def parse_args():
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument('-f', '--file',
|
|
type=str,
|
|
help='Path to the csv or yml file, depends on the mode',
|
|
required=True)
|
|
parser.add_argument('-m', '--mode',
|
|
type=int,
|
|
default=0,
|
|
help='Mode of the script; 0 - one file, 1 - dir')
|
|
parser.add_argument('-i', '--img_prefix',
|
|
type=str,
|
|
default='.',
|
|
help="Prefix in which to save resulted images. default: current pwd")
|
|
parser.add_argument('-y', '--yml_prefix',
|
|
type=str,
|
|
default='.',
|
|
help="Prefix in which to save resulted yaml files. default: current pwd")
|
|
return parser.parse_args()
|