From ed8fbaac3592e6fdfa6cd7321aa29573eb02ef1f Mon Sep 17 00:00:00 2001 From: qqwweee <756134531@qq.com> Date: Sun, 29 Apr 2018 18:31:24 +0800 Subject: [PATCH] add voc_classes --- README.md | 26 +++++++++----------------- model_data/voc_classes.txt | 20 ++++++++++++++++++++ 2 files changed, 29 insertions(+), 17 deletions(-) create mode 100644 model_data/voc_classes.txt diff --git a/README.md b/README.md index a4578afe9..25c54a34d 100644 --- 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 new file mode 100644 index 000000000..8420ab35e --- /dev/null +++ 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