/YOLOv2-TensorFlow

Implement the test part of YOLOv2 with TensorFlow.

Primary LanguagePython

Introduction

This is a repo for implementing YOLOv2 with trained models on tensorflow.

How to use

  • Download the weights file here (OneDrive). The weights file includes 3 files, they should be put in weights folder.
  • The images should be put in test_pic folder, and the video should be put in test_video folder.
  • Then Edit the configure file cfg.py, or use the default configure.
  • Use python YOLO_coco_test.py to run.

Demo

demo
demo

Reference

pjreddie/darknet
gliese581gg/YOLO_tensorflow

Requirements

  • TensorFlow
  • OpenCV