hizhangp / yolo_tensorflow

Tensorflow implementation of YOLO, including training and test phase.

YOLO_tensorflow

Tensorflow implementation of YOLO, including training and test phase.

Installation

1. Clone yolo_tensorflow repository

   $ git clone https://github.com/hizhangp/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/weight

4. Modify configuration in yolo/config.py

5. Training

   $ python train.py

6. Test

   $ python test.py

Requirements

1. Tensorflow

2. OpenCV