Summary

train_ssd.py

@@ -0,0 +1,35 @@
+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 ssd')
+    os.system('cd ssd_keras-master/ && python train.py -c ../' + config_path.json+ ' > ../' + output_path + '/ssd.output 2> ../' + output_path +'/ssd.err')
+    print ('Testing ssd')
+    os.system('cd ssd_keras-master/ && python evaluate.py -c ../' + config_path.json+ ' > ../' + 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)
+