手动训练rknn模型

[复制链接] · 发表于 2022-1-11 14:23:58

请问：是否可以自己训练rknn模型
目的：使用darknet模型训练然后转换rknn 发现模型精度下降，并且有误检或者漏检情况
算法网络：yolov4-tiny-3l
api版本以及驱动版本1.6

自己想手动训练rknn，请问如何训练

只看该作者 · 发表于 2022-1-11 15:18:14

rknn是推理引擎，不支持训练。
转模型的精度下降，在文档中写了通过实时和仿真来排查的方法，建议根据文档逐步排查。

只看该作者 · 发表于 2022-1-11 15:54:29

本帖最后由 czcl 于 2022-1-11 16:00 编辑

模型精度不是下降（我上面的描述存在问题，我的问题是模型转换后，发现一个图像存在误检的情况）
darknet 检测图像精度为0.005
rknn 检测图像精度为0.63

比如检测同一张图，算法为垃圾袋算法
图片里面没有垃圾袋

按照常理，应该检测不出来才对
darknet 没有检测出来，但是降低阈值到0.005的时候检测出来了（这个是合理的）
rknn 检测出来了，检测精度为0.6（不合理）

只看该作者 · 发表于 2022-1-11 17:02:45

这里说的“精度”是指数值精度

只看该作者 · 发表于 2022-1-11 17:44:10

bob286 发表于 2022-1-11 17:02
这里说的“精度”是指数值精度

指的是目标置信度
理论上当图片里面没有垃圾袋的时候，置信度应该比较低才是
但是rknn检测的结果比较高，darknet检测结果正常（代码全部使用官方demo运行）

只看该作者 · 发表于 2022-1-11 18:09:48

czcl 发表于 2022-1-11 17:44
指的是目标置信度
理论上当图片里面没有垃圾袋的时候，置信度应该比较低才是
但是rknn检测的结果比较高， ...

所以要参考文档中的方法来排查数值问题，建议根据文档的指示逐步排查。