Fixed yolo-4c cfg to have 4 classes and added new labels flag

thtrieu · Jun 30, 2017 · 6c65546 · 6c65546
1 parent 47bf990
commit 6c65546
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 3 deletions.
diff --git a/cfg/tiny-yolo-4c.cfg b/cfg/tiny-yolo-4c.cfg
@@ -111,13 +111,13 @@ activation=leaky
 size=1
 stride=1
 pad=1
-filters=50
+filters=45
 activation=linear
 
 [region]
 anchors = 1.08,1.19,  3.42,4.41,  6.63,11.38,  9.42,5.11,  16.62,10.52
 bias_match=1
-classes=5
+classes=4
 coords=4
 num=5
 softmax=1

diff --git a/darkflow/defaults.py b/darkflow/defaults.py
@@ -10,6 +10,7 @@ def setDefaults(self):
         self.define('binary', './bin/', 'path to .weights directory')
         self.define('config', './cfg/', 'path to .cfg directory')
         self.define('dataset', '../pascal/VOCdevkit/IMG/', 'path to dataset directory')
+        self.define('labels', 'labels.txt', 'path to labels file')
         self.define('backup', './ckpt/', 'path to backup folder')
         self.define('summary', './summary/', 'path to TensorBoard summaries directory')
         self.define('annotation', '../pascal/VOCdevkit/ANN/', 'path to annotation directory')

diff --git a/darkflow/net/yolo/misc.py b/darkflow/net/yolo/misc.py
@@ -26,7 +26,7 @@ def labels(meta, FLAGS):
         print("Model has a VOC model name, loading VOC labels.")
         meta['labels'] = labels20
     else:
-        file = 'labels.txt'
+        file = FLAGS.labels
         if model in coco_models:
             print("Model has a coco model name, loading coco labels.")
             file = os.path.join(FLAGS.config, coco_names)