yolov2-tiny 在movidius神经计算棒上运行

1. 在darknet上训练模型

• 项目地址

/srv/users/wenbin.lu/darknet

• 模型地址

/srv/users/wenbin.lu/darknet/backup/yolov2-tiny/20180712/yolov2-tiny-frontal-rear-temp.weights

• cfg文件

/srv/users/wenbin.lu/darknet/cfg/yolov2-tiny-frontal-rear.cfg

2. 将darknet模型转为tensorflow模型

• 项目地址

/srv/users/wenbin.lu/convert/tensorflow_darknet/movidius_tensorflow/yolo-darkflow-movidius

• github地址

https://github.com/fernandodelacalle/yolo-darkflow-movidius

• 模型转化

cd darkflow_all
sudo cp /srv/users/wenbin.lu/darknet/cfg/yolov2-tiny-frontal-rear.cfg tiny-yolo-2class.cfg
sudo cp /srv/users/wenbin.lu/darknet/backup/yolov2-tiny/20180712/yolov2-tiny-frontal-rear.weights tiny-yolo-2class.weights
sudo /srv/users/cong.li/anaconda/envs/tensorflow/bin/python flow --model tiny-yolo-2class.cfg --load  tiny-yolo-2class.weights --savepb
cd built_graph/
ls -l

• tensorflow模型地址

/srv/users/wenbin.lu/convert/tensorflow_darknet/movidius_tensorflow/yolo-darkflow-movidius/darkflow_all/built_graph/tiny-yolo-2class.meta

/srv/users/wenbin.lu/convert/tensorflow_darknet/movidius_tensorflow/yolo-darkflow-movidius/darkflow_all/built_graph/tiny-yolo-2class.pb

3. 将tensorflow模型转为movidius模型

将模型传入装有ncsdk与movidius神经计算棒的树莓派

• 项目地址

/home/pi/Downloads/wenbin.lu/yolov2_frontal_person

• tensorflow模型地址

/home/pi/Downloads/wenbin.lu/yolov2_frontal_person/model_tf

• 将tensorflow模型转为movidius模型

mvNCCompile model_tf/tiny-yolo-2class.pb -s 12 -in input -on output -o model_graph/tiny-yolo-2class.graph 

• movidius模型地址

/home/pi/Downloads/wenbin.lu/yolov2_frontal_person/model_graph/tiny-yolo-2class.graph

• 在神经计算棒上运行模型，对图片input.jpg运行模型，生成output.jpg

cd /home/pi/Downloads/wenbin.lu/yolov2_frontal_person
python3 run_yolov2_frontal_person.py