CAFFE for YOLO9000
Reference
YOLO9000: Better, Faster, Stronger
Usage
caffe
vim Makefile.config
make
Data preparation
cd data/yolo
mkdir lmdb
ln -s /your/path/to/VOCdevkit/ .
python ./get_list.py
# change related path in script convert.sh
./convert.sh
Train
cd examples/yolo/darknet_v3
# change related path in script train.sh
mkdir models
./train_darknet_v3.sh
Test a image
cd examples/yolo/eval_detection
jupyter notebook
test_det.ipynb
Eval VOCtest2007(The first way online)
# mAP reach ~56. Because of I train net poorly...you can try.
cd examples/yolo/darknet_v3
./test_darknet_v3.sh
MODEL=./models/gnet_yolo_region_darknet_v3_pretrain_rectify_iter_200000.caffemodel
model is here:
Eval VOCtest2007(The second way offline)
cd examples/eval_detection
python test_yolo_v2.py
Draw loss figure(avg_obj, avg_noobj, avg_class, avg_iou, recall)
cd tools/yolo_extra
python parse_log_yolo.py ./log/train_darknet_anchor.log ./log