Home

Awesome

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