Skip to content

chl916185/caffe-yolo

 
 

Repository files navigation

CAFFE for YOLO

Reference

You Only Look Once: Unified, Real-Time Object detection

http://arxiv.org/abs/1506.02640

http://www.yeahkun.com/2016/09/06/object-detection-you-only-look-once-caffe-shi-xian/

Usage

Data preparation

  cd data/yolo
  ln -s /your/path/to/VOCdevkit/ .
  python ./get_list.py
  # change related path in script convert.sh
  ./convert.sh 

Train

  cd examples/yolo
  # change related path in script train.sh
  mkdir models
  nohup ./train.sh &

Test

  # if everything goes well, the map of gnet_yolo_iter_32000.caffemodel may reach ~56.
  cd examples/yolo
  ./test.sh model_path gpu_id
  The model is here (link: https://pan.baidu.com/s/1jHAN6xK password: kvee)

About

Caffe for YOLO

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • C++ 80.8%
  • Python 8.8%
  • Cuda 5.6%
  • CMake 2.7%
  • MATLAB 0.9%
  • Makefile 0.7%
  • Other 0.5%