YOLO_v1

This implementation of YOLO.

Installation

1. Clone YOLO_v1 repository

   $ git clone https://github.com/leeyoshinari/YOLO_v1.git
   $ cd YOLO_v1

2. Download Pascal VOC2007 dataset, and put the dataset into data/Pascal_voc.

   if you download other dataset, you also need to modify file paths.

3. Download weights file YOLO_small, and put weight file into data/output

   Or you can also download my training weights file YOLO_v1.

4. Modify configuration into yolo/config.py

5. Training

   $ python train.py

6. Test

   $ python test.py

Training on Your Own Dataset

To train the model on your own dataset, you should need to modefy:

1. Put all the images into the Images folder, put all the labels into the Labels folder. Select a part of the images for training, write this part of the images filename into train.txt, the remaining part of the images filename written in test.txt. Then put the Images, Labels, train.txt and test.txt into data/data_set. Put weight file in data/output

2. config.py modify the CLASSES

3. train.py replacefrom utils.pascal_voc import pascal_voc with from utils.preprocess import preprocess, and replace pascal = pascal_voc() with pascal = preprocess().

Requirements

1. Tensorflow
2. OpenCV