36 lines
976 B
Python
36 lines
976 B
Python
import os
|
|
import argparse
|
|
|
|
|
|
def makedirs(path):
|
|
try:
|
|
os.makedirs(path)
|
|
except OSError:
|
|
if not os.path.isdir(path):
|
|
raise
|
|
|
|
|
|
|
|
|
|
|
|
def _main_(args):
|
|
|
|
config_path = args.conf
|
|
output_path = args.output
|
|
|
|
makedirs(output_path)
|
|
|
|
print ('Training ssd')
|
|
os.system('cd ssd_keras-master/ && python train.py -c ../' + config_path + ' > ../' + output_path + '/ssd.output 2> ../' + output_path +'/ssd.err')
|
|
print ('Testing ssd')
|
|
os.system('cd ssd_keras-master/ && python evaluate.py -c ../' + config_path + ' > ../' + output_path + '/ssd_test.output 2> ../' + output_path +'/ssd_test.err')
|
|
|
|
|
|
if __name__ == '__main__':
|
|
argparser = argparse.ArgumentParser(description='train and evaluate ssd model on any dataset')
|
|
argparser.add_argument('-c', '--conf', help='path to configuration file')
|
|
argparser.add_argument('-o', '--output', help='path to save the experiment')
|
|
args = argparser.parse_args()
|
|
_main_(args)
|
|
|