YOLO-tensorflow

Tensorflow implementation of YOLO, including training and test phase.

Installation

1. Clone YOLO-tensorflow repository

   $ git clone https://github.com/sungjunhong/YOLO-tensorflow.git
   $ cd YOLO-tensorflow

2. Download Pascal VOC dataset, and create correct directories

   $ ./download_data.sh

3. Download YOLO_small weight file and put it in data/weights

4. Modify configuration in yolo/config.py

5. Training

   $ python train.py

6. Test

   $ python test.py

Requirements

1. Tensorflow

2. OpenCV