Toybrick

手动训练rknn模型

czcl

中级会员

积分
365
发表于 2022-1-11 14:23:58    查看: 2473|回复: 5 | [复制链接]    打印 | 显示全部楼层
请问:是否可以自己训练rknn模型
目的:使用darknet模型训练然后转换rknn 发现模型精度下降,并且有误检或者漏检情况
算法网络:yolov4-tiny-3l
api版本以及驱动版本1.6

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



回复

使用道具 举报

bob286

中级会员

积分
323
发表于 2022-1-11 15:18:14 | 显示全部楼层
rknn是推理引擎,不支持训练。
转模型的精度下降,在文档中写了通过实时和仿真来排查的方法,建议根据文档逐步排查。
回复

使用道具 举报

czcl

中级会员

积分
365
 楼主| 发表于 2022-1-11 15:54:29 | 显示全部楼层
本帖最后由 czcl 于 2022-1-11 16:00 编辑

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

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

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

使用道具 举报

bob286

中级会员

积分
323
发表于 2022-1-11 17:02:45 | 显示全部楼层
这里说的“精度”是指数值精度
回复

使用道具 举报

czcl

中级会员

积分
365
 楼主| 发表于 2022-1-11 17:44:10 | 显示全部楼层
bob286 发表于 2022-1-11 17:02
这里说的“精度”是指数值精度

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

使用道具 举报

bob286

中级会员

积分
323
发表于 2022-1-11 18:09:48 | 显示全部楼层
czcl 发表于 2022-1-11 17:44
指的是目标置信度
理论上当图片里面没有垃圾袋的时候,置信度应该比较低才是
但是rknn检测的结果比较高, ...

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

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

产品中心 购买渠道 开源社区 Wiki教程 资料下载 关于Toybrick


快速回复 返回顶部 返回列表