33 lines
1.0 KiB
Python
33 lines
1.0 KiB
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 full_yolo3')
|
|
os.system('cd keras-yolo3-master/ && python train.py -c ../' + config_path + ' > ../' + output_path + '/yolo3_full_yolo.output 2> ../' + output_path +'/yolo3_full_yolo.err')
|
|
print('Test full_yolo3')
|
|
os.system('cd keras-yolo3-master/ && python evaluate.py -c ../' + config_path+ ' > ../' + output_path + '/yolo3_full_yolo_test.output 2> ../' + output_path +'/yolo3_full_yolo_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)
|