https://github.com/choasup/caffe-yolo9000
Caffe for YOLOv2 & YOLO9000
https://github.com/choasup/caffe-yolo9000
caffe detection
Last synced: 5 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 (about 8 years ago)
- Default Branch: master
- Last Pushed: 2018-07-02T07:49:40.000Z (almost 7 years ago)
- Last Synced: 2024-11-09T14:38:17.776Z (6 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
Awesome Lists containing this project
- awesome-yolo-object-detection - choasup/caffe-yolo9000 - yolo9000?style=social"/> : Caffe for YOLOv2 & YOLO9000. (Other Versions of YOLO)
- 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
```