/yolov2-tensorflow

Primary LanguagePythonMIT LicenseMIT

yolov2_tensorflow

Requirements

  1. Tensorflow
  2. OpenCV

Tensorflow implementation of YOLO, including yolov1 and yolov2 demo.

Installation

  1. Clone yolov2_tensorflow repository

    $ git clone https://github.com/shishichang/yolov2-tensorflow.git
$ cd yolov2_tensorflow

  2. Download YOLO_v1 YOLO_v2_pb YOLO_v2_meta put it in models

  3. Modify configuration in yolo/config.py for yolov1

  4. Run

    $ python demo_yolo_v1.py
$ python demo_yolo_v2.py