Toybrick

手动训练rknn模型

czcl

中级会员

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

自己想手动训练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(不合理)
回复

使用道具 举报

czcl

中级会员

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

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

使用道具 举报

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

本版积分规则

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


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