- caffe training
- add pre-trained model
- fix bugs
- windows support
- vehicle detection
- Visual Studio 2013 or 2015
- CMake 3.4 or higher (Visual Studio and Ninja generators are supported)
- Anaconda
The build step was the same as MobileNet-SSD-windows
> cd $caffe_root/script
> build_win.cmd
config build_win.cmd "CPU_ONLY" = 0
cd examples
- convert yolo.cfg to yolo.prototxt
- convert yolo weights to caffemodel
cd caffe_root
- Download pre-trained caffmodel , save at location $caffe_root/models/yolo/
- examples\yolo_detection.cmd
If load success , you can see the image window like this
There has two ways for training your dataset
- training with caffe (recommand)
- training with darknet project and convert the weights to caffemodel
Save lmdb at location $caffe_root\data\yolo\
Put your dataset into $caffe_root\data\yolo\VOCdevkit
> cd $caffe_root\data\yolo
> python get_list.py
Check files label_map.txt,trainval.txt,test_2007.txt are all correct
> cd $caffe_root\data\yolo
> create_dataset.cmd
> cd $caffe_root
> examples\yolo_train.cmd
- There has no data augmentation code inside (ex.hue,rotation) , note the training mAp was bad
- I still try to train a good caffemodel , currently ,the deploy model was just show how to do training
- Add MobileNet-YOLO
- Data augmentation