Toybrick

关于darknet-yolov4 转换成rknn模型的问题

SimonNeko

新手上路

积分
33
发表于 2021-1-26 20:54:50    查看: 13817|回复: 7 | [复制链接]    打印 | 显示全部楼层
from rknn.api import RKNN

rknn = RKNN(verbose=True)
print("rknn initialized")
rknn.config(batch_size=1, channel_mean_value='0 0 0 255', reorder_channel='0 1 2', need_horizontal_merge=True)
print("config finish")
rknn.load_darknet("yolov4.cfg", "yolov4.weights")
print("load darknet model")
rknn.build(do_quantization=False)
print("build rknn model")
rknn.export_rknn("./yolo_v4.rknn")
print("export rknn model")
rknn.release()
请问这样对yolov4的模型进行转换是否正确?望解答
回复

使用道具 举报

SimonNeko

新手上路

积分
33
 楼主| 发表于 2021-2-1 12:56:26 | 显示全部楼层
本帖最后由 SimonNeko 于 2021-2-1 13:51 编辑

from rknn.api import RKNN

rknn = RKNN(verbose=True)
print("rknn initialized")
rknn.config(batch_size=1, channel_mean_value='0 0 0 255', reorder_channel='2 1 0', need_horizontal_merge=False)
print("config finish")
rknn.load_darknet("yolov3.cfg", "yolov3.weights")
print("load darknet model")
rknn.build(do_quantization=False)
print("build rknn model")
rknn.export_rknn("./yolov3.rknn")
print("export rknn model")
rknn.release()

我用如上代码, 将darknet的yolov3转换成了rknn模型, 使用flask部署在了rk1808上



图片使用的是下面的这张, 在输入模型之前转成了416*416*3

下面这一步需要大约3s的时间才能执行完
outputs = rknn.inference(inputs=[image_mat_BGR])

请问这正常吗?
回复

使用道具 举报

SimonNeko

新手上路

积分
33
 楼主| 发表于 2021-2-1 13:51:20 | 显示全部楼层

本帖子中包含更多资源

您需要 登录 才可以下载或查看,没有帐号?立即注册

x
回复

使用道具 举报

jefferyzhang

版主

积分
12844
发表于 2021-2-2 14:19:36 | 显示全部楼层
rknn模型的具体转换问题,都可以直接加入置顶帖rknn官方qq群进行询问和探讨
回复

使用道具 举报

SimonNeko

新手上路

积分
33
 楼主| 发表于 2021-2-10 01:04:46 | 显示全部楼层
jefferyzhang 发表于 2021-2-2 14:19
rknn模型的具体转换问题,都可以直接加入置顶帖rknn官方qq群进行询问和探讨

好的, 感谢~
回复

使用道具 举报

brunolin

高级会员

积分
611
发表于 2021-3-5 09:24:35 | 显示全部楼层
SimonNeko 发表于 2021-2-1 12:56
from rknn.api import RKNN

rknn = RKNN(verbose=True)

rknn.build(do_quantization=False)
你沒有做量化,推理時間長是正常的
要在板子上執行必須做量化再進行推理才會加速
回复

使用道具 举报

zhaohuimima

注册会员

积分
73
发表于 2021-3-10 18:08:09 | 显示全部楼层
请问楼主,
你在转换的时候,cfg文件和weight文件修改过吗?我用官方下载的cfg和weights文件,转换后inference的结果不对。小目标的向量在最前面。
回复

使用道具 举报

zhaohuimima

注册会员

积分
73
发表于 2021-3-10 18:10:28 | 显示全部楼层
比如,原本的输出应该是【13*13】[26*26][52*52],我这里的结果是反过来的【52*52】[26*26]【13*13】。
回复

使用道具 举报

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

本版积分规则

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


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