add voc_classes

Roscript · Apr 29, 2018 · ed8fbaa · ed8fbaa
1 parent fda7b44
commit ed8fbaa
Show file tree

Hide file tree

Showing 2 changed files with 29 additions and 17 deletions.
diff --git a/README.md b/README.md
@@ -24,20 +24,12 @@ python yolo.py   OR   python yolo_video.py
 
 ## Training
 
-1. Generate your own annotation file.
-
-  One row for one image;
-
-  Row format: image_file_path box1 box2 ... boxN;
-
-  Box format: x_min,y_min,x_max,y_max,class_id (no space).
-
-  For VOC dataset, try `python voc_annotation.py`
-
-2. Generate your own class names file.
-
-3. Modify train.py and start training.
-
-  `python train.py`
-
-  You will get the trained model model_data/my_yolo.h5.
+1. Generate your own annotation file and class names file.  
+    One row for one image;  
+    Row format: image_file_path box1 box2 ... boxN;  
+    Box format: x_min,y_min,x_max,y_max,class_id (no space).  
+    For VOC dataset, try `python voc_annotation.py`
+
+2. Modify train.py and start training.  
+    `python train.py`  
+    You will get the trained model model_data/my_yolo.h5.
diff --git a/model_data/voc_classes.txt b/model_data/voc_classes.txt
@@ -0,0 +1,20 @@
+aeroplane
+bicycle
+bird
+boat
+bottle
+bus
+car
+cat
+chair
+cow
+diningtable
+dog
+horse
+motorbike
+person
+pottedplant
+sheep
+sofa
+train
+tvmonitor