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