linuxでプログラムを実行するときにパラメータを追加する
4357 ワード
import argparse
parser = argparse.ArgumentParser(description='Defence')
parser.add_argument('--input_dir', metavar='DIR', default='',
help='Input directory with images.')
parser.add_argument('--output_file', metavar='FILE', default='',
help='Output file to save labels.')
parser.add_argument('--img-size', type=int, default=299, metavar='N',
help='Image patch size (default: 299)')
parser.add_argument('--no-gpu', action='store_true', default=False,
help='disables GPU training')
def main():
args = parser.parse_args()
if not os.path.exists(args.input_dir):
print("Error: Invalid input folder %s" % args.input_dir)
exit(-1)