Skip to content

Latest commit

 

History

History
38 lines (27 loc) · 790 Bytes

File metadata and controls

38 lines (27 loc) · 790 Bytes

YOLO_tensorflow

Taken from this github repo!

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/weights

  4. Modify configuration in yolo/config.py

  5. Training

    $ python train.py
  6. Test

    $ python test.py

Requirements

  1. Tensorflow

  2. OpenCV