darknet yolov3 yolov4转caffe 教程

最新推荐文章于 2023-08-02 11:28:24 发布

鲤鱼不懂

最新推荐文章于 2023-08-02 11:28:24 发布

阅读量1.2k

点赞数 1

分类专栏： darknet yolov3 caffe 文章标签： caffe 人工智能深度学习

本文链接：https://blog.csdn.net/qq_43318374/article/details/121058146

版权

darknet yolov3 caffe 专栏收录该内容

2 篇文章 2 订阅

订阅专栏

在目标检测任务中，我们常使用yolov3, yolov4作为目标检测模型，在实际部署工作中，我们很多边缘设备需要把模型转换成caffe。在次记录下转换过程。
　　我们需要在caffe框架中，添加所需要的层。这个转换工具可以参考github上一位大佬项目：

git clone https://github.com/ChenYingpeng/darknet2caffe.git

把大佬mish_layer文件中的mish_layer.hpp和upsample_layer文件中的upsample_layer.hpp复制到安装caffe目录下的include/caffe/layers/文件夹中。

├── mish_layer
│   ├── mish_layer.cpp
│   ├── mish_layer.cu
│   └── mish_layer.hpp
├── pooling_layer
│   └── pooling_layer.cpp
└── upsample_layer
    ├── upsample_layer.cpp
    ├── upsample_layer.cu
    └── upsample_layer.hpp

把大佬mish_layer文件中的mish_layer.cpp、mish_layer.cu 和　upsample_layer文件中的
upsample_layer.cpp、upsample_layer.cu复制到caffe目录下的/src/caffe/layers/文件夹中。
最后在caffe目录下的/src/caffe/proto/caffe.proto。打开这个文件，添加以下内容：
在423行下面添加：

optional UpsampleParameter upsample_param = 149;
optional MishParameter mish_param = 150;

在这里插入图片描述

在最后一行下添加如下内容：

message UpsampleParameter{

optional int32 scale = 1 [default = 1];

}

message MishParameter {
  enum Engine {
    DEFAULT = 0;
    CAFFE = 1;
    CUDNN = 2;
  }
  optional Engine engine = 2 [default = DEFAULT];
}

在这里插入图片描述
然后我们重新编译一下caffe，在caffe/build目录下

sudo make clean
sudo make all -j8
sudo make pycaffe -j8

最后执行转换命令

python3 darknet2caffe.py

打印信息如下

I1030 22:34:06.650609  7437 net.cpp:202] layer1-conv does not need backward computation.
I1030 22:34:06.650612  7437 net.cpp:202] input does not need backward computation.
I1030 22:34:06.650614  7437 net.cpp:244] This network produces output layer139-conv
I1030 22:34:06.650619  7437 net.cpp:244] This network produces output layer150-conv
I1030 22:34:06.650621  7437 net.cpp:244] This network produces output layer161-conv
I1030 22:34:06.650758  7437 net.cpp:257] Network initialization done.
unknow layer type yolo 
unknow layer type yolo 
save prototxt to ./yolov4.protofile
save caffemodel to ./yolov4.caffemode