yolo_tensorflow

yolo_tensorflow copied to clipboard

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