This project shows how to run tiny yolov2 (20 classes) with movidius stick:
The following experiments are done on an Intel NUC with ubuntu 16.04.
Please install NCSDK following https://github.com/movidius/ncsdk.
make
mvNCCompile ./models/caffemodels/yoloV2Tiny20.prototxt -w ./models/caffemodels/yoloV2Tiny20.caffemodel -s 12
There will be a file graph generated as converted models for NCS.
python3 ./detectionExample/Main.py --image ./data/dog.jpg
This loads graph by default and results will be like this:
Install caffe and config the python environment path.
sh ./models/convertyo.sh
Tips:
Please ignore the error message similar as "Region layer is not supported".
The converted caffe models should end with "prototxt" and "caffemodel".
Please update parameters (biases, object names, etc) in ./src/CRegionLayer.cpp, and parameters (dim, blockwd, targetBlockwd, classe, etc) in ./detectionExample/ObjectWrapper.py.
Please read ./src/CRegionLayer.cpp and ./detectionExample/ObjectWrapper.py for details.
Research Only
duangenquan@gmail.com