Toybrick

yolov4的模型推理结果

clover

新手上路

积分
24
楼主
发表于 2020-4-30 11:04:17 | 显示全部楼层
我尝试转了一下,就只有mish层不支持,但转换也不会报错,只是会把mish层转成relu层,所以推理结果不正确。我把mish层换成leaky relu之后转换推理结果都是正常的。希望能添加支持mish激活函数。
回复

使用道具 举报

clover

新手上路

积分
24
沙发
发表于 2020-4-30 11:27:22 | 显示全部楼层
jefferyzhang 发表于 2020-4-30 11:16
大家可以先把yolov4的 mish都改relu来训练,看下效果。
mish这个激活函数支持起来估计需要点时间,也不利于 ...

我实验了把mish换成leaky relu,精度还是有下降,所以还是希望能支持mish。
回复

使用道具 举报

clover

新手上路

积分
24
板凳
发表于 2020-5-7 10:24:56 | 显示全部楼层
qq1993106 发表于 2020-5-6 11:04
直接cfg修改后训练的?大概有多少精度损失能告知一下吗?

对,修改cfg把mish全部换成leaky。精度在我自己的数据集上测试下来,map@0.75大概掉了5个点
回复

使用道具 举报

clover

新手上路

积分
24
地板
发表于 2020-5-7 14:23:13 | 显示全部楼层
yanggui 发表于 2020-5-7 13:56
你好能不能把你转换后的yolo4共享出来看看,你调用yolo4转换后的模型的检测程序用哪个啊? ...

模型和代码我这边不方便公开。转换的话只要把darknet训练完的cfg和weight文件,用rknn_toolkit示例中的darknet例子代码修改部分参数就可以转换,转完后可以先用例子中的python接口模拟看看结果对不对,后处理部分需要修改一下anchor的值和其他一些参数。转换完后推理调用的代码参照rknn api当中的例子写,就是后处理部分有些复杂,可以对照python的代码慢慢改。
回复

使用道具 举报

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

本版积分规则

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


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