海思hi3516dv300芯片AI图像模块学习（五）Darknet模型转Caffe模型

之前在DarkNet框架下训练过YOLOV3模型，由于官方教程YOLO也是基于darknet框架，先拿这个模型试试水。

参考：https://gitee.com/openharmony/device_soc_hisilicon/blob/master/hi3516dv300/sdk_linux/sample/taurus/doc/4.2.4.%E6%A3%80%E6%B5%8B%E7%BD%91.md#4243-darknet%E6%A8%A1%E5%9E%8B%E8%BD%ACcaffe%E6%96%B9%E6%A1%88

cd code

git clone https://gitee.com/losschen/darknet2caffe.git

prototxt.py修改方法，官方写的不详细，写了前两部分

参考：https://blog.csdn.net/weixin_42048023/article/details/106021536，改第三部分

 def print_block(block_info, prefix, indent):

接着改:

cd ~/caffe/src/caffe/proto/

gedit caffe.proto

注意路径要对

cp caffe_layers/upsample_layer/upsample_layer.hpp /home/keii/caffe/include/caffe/layers/

cp caffe_layers/upsample_layer/upsample_layer.c*  /home/keii/caffe/include/caffe/layers/

拷贝进去后要重新编译caffe,等于把刚加进去的层也编译下

make clean

make -j4

make pycaffe

没报错。

yolov3 darknet模型 放到darknet2caffe文件夹

python3 darknet2caffe.py yolov3_darknet/yolov3.cfg  yolov3_darknet/yolov3_final.weights yolov3_caffe/yolov3_caffe.prototxt  yolov3_caffe/yolov3_caffe.caffemodel

readme 里面的sample命令没有 darknet2caffe.py，一定要加进去。

转化成caffe模型成功!