Summary

train_yolo.py

@@ -0,0 +1,32 @@
+import os
+
+
+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.json+ ' > ../' + 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.json+ ' > ../' + 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)