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 时间: 2020-5-20 15:14

jefferyzhang 发表于 2020-5-15 16:09
我也不知道

应该快出来了吧，出来后rk板子（3399pro或1808）上是不是直接update就可以了哈

作者: jefferyzhang 时间: 2020-5-20 16:04

Devin 发表于 2020-5-20 15:14
应该快出来了吧，出来后rk板子（3399pro或1808）上是不是直接update就可以了哈 ...

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

欢迎光临 Toybrick (https://t.rock-chips.com/)