Toybrick

标题: yolov4的模型推理结果 [打印本页]

作者: qq1993106    时间: 2020-4-28 09:04
标题: yolov4的模型推理结果
官方放出来了yolov4,源码主页:https://github.com/AlexeyAB/darknet,尝试用rknn1.3.0进行模型转化(不量化),尝试进行推理结果异常。
所以希望确认一下yolov4各个layer是否支持rknn推理。
yolov4官方cfg:见附件 yolov4.cfg
yolov4转rknn的log: 见附件 verbose.log


作者: jefferyzhang    时间: 2020-4-28 10:29
已经上报了
作者: qq1993106    时间: 2020-4-28 13:50
jefferyzhang 发表于 2020-4-28 10:29
已经上报了

感谢,期待答复!
作者: clover    时间: 2020-4-30 11:04
我尝试转了一下,就只有mish层不支持,但转换也不会报错,只是会把mish层转成relu层,所以推理结果不正确。我把mish层换成leaky relu之后转换推理结果都是正常的。希望能添加支持mish激活函数。
作者: jefferyzhang    时间: 2020-4-30 11:16
大家可以先把yolov4的 mish都改relu来训练,看下效果。
mish这个激活函数支持起来估计需要点时间,也不利于NPU上跑到最大性能。。。
作者: clover    时间: 2020-4-30 11:27
jefferyzhang 发表于 2020-4-30 11:16
大家可以先把yolov4的 mish都改relu来训练,看下效果。
mish这个激活函数支持起来估计需要点时间,也不利于 ...

我实验了把mish换成leaky relu,精度还是有下降,所以还是希望能支持mish。
作者: yanggui    时间: 2020-4-30 17:35
四楼的你怎么把mish层换成leaky relu层啊?分析一下经验啊,
作者: qq1993106    时间: 2020-5-6 11:04
clover 发表于 2020-4-30 11:27
我实验了把mish换成leaky relu,精度还是有下降,所以还是希望能支持mish。

直接cfg修改后训练的?大概有多少精度损失能告知一下吗?
作者: clover    时间: 2020-5-7 10:24
qq1993106 发表于 2020-5-6 11:04
直接cfg修改后训练的?大概有多少精度损失能告知一下吗?

对,修改cfg把mish全部换成leaky。精度在我自己的数据集上测试下来,map@0.75大概掉了5个点
作者: yanggui    时间: 2020-5-7 13:56
clover 发表于 2020-5-7 10:24
对,修改cfg把mish全部换成leaky。精度在我自己的数据集上测试下来,大概掉了5个点 ...

你好能不能把你转换后的yolo4共享出来看看,你调用yolo4转换后的模型的检测程序用哪个啊?
作者: clover    时间: 2020-5-7 14:23
yanggui 发表于 2020-5-7 13:56
你好能不能把你转换后的yolo4共享出来看看,你调用yolo4转换后的模型的检测程序用哪个啊? ...

模型和代码我这边不方便公开。转换的话只要把darknet训练完的cfg和weight文件,用rknn_toolkit示例中的darknet例子代码修改部分参数就可以转换,转完后可以先用例子中的python接口模拟看看结果对不对,后处理部分需要修改一下anchor的值和其他一些参数。转换完后推理调用的代码参照rknn api当中的例子写,就是后处理部分有些复杂,可以对照python的代码慢慢改。
作者: auuuux    时间: 2020-5-7 22:08
clover 发表于 2020-5-7 10:24
对,修改cfg把mish全部换成leaky。精度在我自己的数据集上测试下来,大概掉了5个点 ...

楼主,方便把修改后的cfg 文件放出来看看么? 或者给个示例
作者: jefferyzhang    时间: 2020-5-15 10:53
NPU Team 回复:

mish函数已支持,等待下个版本1.3.3发布。
作者: tomyhome    时间: 2020-5-15 15:59
jefferyzhang 发表于 2020-5-15 10:53
NPU Team 回复:

mish函数已支持,等待下个版本1.3.3发布。


1.3.3版本什么时候发布呢?能有个时间点吗?
作者: jefferyzhang    时间: 2020-5-15 16:09
tomyhome 发表于 2020-5-15 15:59
1.3.3版本什么时候发布呢?能有个时间点吗?

我也不知道
作者: Devin    时间: 6 天前
jefferyzhang 发表于 2020-5-15 16:09
我也不知道

应该快出来了吧,出来后rk板子(3399pro或1808)上是不是直接update就可以了哈
作者: jefferyzhang    时间: 6 天前
Devin 发表于 2020-5-20 15:14
应该快出来了吧,出来后rk板子(3399pro或1808)上是不是直接update就可以了哈 ...

是的,出来后论坛置顶贴会发公告,然后直接update/upgrade就可以。




欢迎光临 Toybrick (http://t.rock-chips.com/) Powered by Discuz! X3.3