Darknet2Caffe 大概步骤

本文介绍了如何将Darknet训练的模型转换为Caffe格式。首先在Linux服务器上训练Yolov3模型,接着详细阐述Caffe的安装过程,包括解决安装问题和添加upsample层。然后通过darknet2caffe工具转换模型,转换过程中需手动处理yolo层。最后,转换完成后需自行实现yolo层的后处理代码。
摘要由CSDN通过智能技术生成

注:第四、五、六节文章只是极端理想的情况(模型移植精度损失不大,后处理代码也无需过多修改),可以作为熟悉大致流程的例子看看。移植模型绝对不会这么理想…

1训练模型

我的模型训练是在 Linux 服务器上进行。因为我的笔记本没有 GPU。
使用 yolov3 训练模型网上有很多教程。大体上就是先标注数据再训练得到模型。
大家查一下就可以看到很多图文并茂的教程了。我就不重复了。
完事之后,你会得到如下2个文件:

xxx.cfg
xxx.weights

2转换模型

因为相机需要的是 WK 模型,而 WK 模型只能从 Caffe 模型转换得到。
所以我们先把 Darknet 转换到 Caffe。

2.1安装Caffe

第一步当然是安装 Caffe。
Caffe 的安装非常坑,光是安装就可以写一个博客了。
网上也有很多教程,每个人遇到的问题也不一样,这个只能靠自己(Google)了。
我只能说一说大体上的步骤。
你可能需要先安装 protobuf-2.6.1
需要的Caffe源码 :https://github.com/BVLC/caffe
编辑Makefile.config之后是编译安装。

make all
make t

  • 1
    点赞
  • 8
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值