/caffe-yolo

Caffe for YOLO

Primary LanguageC++OtherNOASSERTION

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)