Tensorflow implementation of YOLO, including training and test phase.
Switch branches/tags
Nothing to show
Clone or download
Latest commit 88aba9d Feb 28, 2018

README.md

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