35 lines
943 B
Python
35 lines
943 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('python ssd_keras-master/train.py -c ' + config_path + ' > ' + output_path + '/ssd.output 2> ' + output_path +'/ssd.err')
|
|
print ('Testing ssd')
|
|
os.system('python ssd_keras-master/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)
|