xingwangsfu / caffe-yolo

YOLO (Real-Time Object Detection) in caffe
520 stars 336 forks source link

# YOLO in caffe

Update 12-05-2016: Currently, only yolo v1 (http://pjreddie.com/darknet/yolov1/) is supported. Yolo V2 (http://pjreddie.com/darknet/yolo/) is not supported. Batch norm layer is supported.

Introduction

This is a caffe implementation of the YOLO:Real-Time Object Detection

Note, the caffe models are not trained in caffe, but converted from darknet's (.weight) files (http://pjreddie.com/darknet/yolov1/).

The converter is consisted of four steps:

Usage

Convert yolo's (.weight) files to caffemodel

Main file usage

run yolo_main.py to do yolo object detection for the input image

replace model_filename with /your/path/to/yolo_small_deploy.prototxt or yolo_tiny_deploy.prototxt, weight_filename with /your/path/to/yolo_tiny.caffemodel or yolo_small.caffemodel and image_filename with the target image file

Requirements

Copyrights

According to the LICENSE file of the original code,