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

博客介绍了搭建caffe环境的方法,指出Yolov3中部分层caffe不支持,如shortcut可用eltwise替换、route可用concat替换等。还给出添加upsample层的代码及添加过程,编译成功后可加载yolov3相关文件运行,代码基于c/c++,Windows搭建环境后也可使用。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

https://blog.youkuaiyun.com/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++的,可以直接编译使用,自己已经测试过;

评论 5
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值