https://github.com/choasup/caffe-yolo9000

Caffe for YOLOv2 & YOLO9000
https://github.com/choasup/caffe-yolo9000

caffe detection

Last synced: 12 days ago
JSON representation

Caffe for YOLOv2 & YOLO9000

• Host: GitHub
• URL: https://github.com/choasup/caffe-yolo9000
• Owner: choasup
• License: other
• Created: 2017-03-01T12:24:38.000Z (over 7 years ago)
• Default Branch: master
• Last Pushed: 2018-07-02T07:49:40.000Z (almost 6 years ago)
• Last Synced: 2024-02-29T07:36:21.154Z (4 months ago)
• Topics: caffe, detection
• Language: C++
• Homepage:
• Size: 55.5 MB
• Stars: 166
• Watchers: 18
• Forks: 134
• Open Issues: 6
• Metadata Files:
  • Readme: README.md
  • Contributing: CONTRIBUTING.md
  • License: LICENSE

Lists

• awesome-yolo-object-detection - choasup/caffe-yolo9000 - yolo9000?style=social"/> : Caffe for YOLOv2 & YOLO9000. (Other Versions of YOLO)

README

        
# CAFFE for YOLO9000

## Reference

> YOLO9000: Better, Faster, Stronger

> http://pjreddie.com/yolo9000/

> https://github.com/yeahkun/caffe-yolo

> https://github.com/weiliu89/caffe/tree/ssd

## Usage

### Caffe 

```Shell

   vim Makefile.config

   make clean

   make all -j8

   make pycaffe

```

(Be careful: caffe&caffe2 PYTHONPATH conflict)

### Data preparation

Like SSD data setting.

```Shell

  cd data/VOC0712

  

  vim create_data.sh

  ./create_data.sh 

```

### Train

```Shell

  cd examples/yolo/darknet_voc

  mkdir models

  

  # edit according yourself

  vim gnet_region_train_darknet448_voc.prototxt

  vim gnet_region_test_darkent448_voc.prototxt

  vim gnet_region_solver_darkent448_voc.prototxt

  # download pretrain_model

```  

  > https://pan.baidu.com/s/1c71EB-6A1xQb2ImOISZiHA password: 9u5v

```

  # change related path in script train.sh

  vim train_darknet448.sh

  

  ./train_darknet448.sh

```

### Eval mAP

```Shell

   cd demo

   vim demo_yolo.py

   python demo_yolo.py

```

### Test a image

```Shell

   vim test_darknet448.sh

   ./test_darknet448.sh

```