Merge pull request #188 from abagshaw/master

Fix file path for YOLOv1 training
rjdbcm · Apr 26, 2017 · 1dcd74e · 1dcd74e
2 parents 4ca6969 + 7436d3f
commit 1dcd74e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/darkflow/net/yolo/data.py b/darkflow/net/yolo/data.py
@@ -14,7 +14,7 @@ def parse(self, exclusive = False):
     """
     meta = self.meta
     ext = '.parsed'
-    history = os.path.join('net', 'yolo', 'parse-history.txt');
+    history = os.path.join('darkflow', 'net', 'yolo', 'parse-history.txt');
     if not os.path.isfile(history):
         file = open(history, 'w')
         file.close()
@@ -36,7 +36,7 @@ def parse(self, exclusive = False):
     print('\n{} parsing {}'.format(meta['model'], ann))
     dumps = pascal_voc_clean_xml(ann, meta['labels'], exclusive)
 
-    save_to = os.path.join('net', 'yolo', meta['name'])
+    save_to = os.path.join('darkflow', 'net', 'yolo', meta['name'])
     while True:
         if not os.path.isfile(save_to + ext): break
         save_to = save_to + '_'