基于caffe框架复现yolov3目标检测

https://blog.csdn.net/Chen_yingpeng/article/details/80692018

搭建caffe环境

    首先caffe环境搭建自行百度解决,其次需要了解Yolov3里面有shortcut、route、upsample、yolo等这些层是caffe不支持的,但是shortcut可以用eltwise替换,route可以用concat替换,yolo只能自己写,upsample可以添加。这里添加upsample这一层的代码,代码地址:链接:https://pan.baidu.com/s/13GpoYoqKSCeFX0m0ves_fQ 密码:bwrd

    添加过程:把upsample_layer.hpp 放在include/caffe/layers下面;

                     把upsample_layer.cpp与upsample_layer.cu放在src/caffe/layers下面;

    往自己的caffe.proto里面添加UpsampleParameter相关参数;可以参考我上传的caffe.proto;

然后重新编译;编译成功之后就可以加载yolov3.prototxt与yolov3.caffemodel运行;

代码下载地址:https://github.com/ChenYingpeng/caffe-yolov3

补充说明:Windows也可以只要搭建了相应的环境,代码都是基于c/c++的,可以直接编译使用,自己已经测试过;

你可能感兴趣的:(编程,深度学习)