之前在DarkNet框架下訓(xùn)練過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修改方法，官方寫的不詳細(xì)，寫了前兩部分

參考：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/

拷貝進(jìn)去后要重新編譯caffe,等于把剛加進(jìn)去的層也編譯下

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，一定要加進(jìn)去。

轉(zhuǎn)化成caffe模型成功!