Toybrick

RKNN能否推理NV12图片

zhuo799154051

中级会员

积分
443
楼主
发表于 2020-10-9 12:04:04    查看: 7220|回复: 1 | [复制链接]    打印 | 只看该作者
我本想根据官方demo将解码后的RGB24图像进行推理和打框,然后将处理后的图像编码为H264并保存。
但是我将把demo改为 EncCtx ctx; ctx.v4l2format = V4L2_PIX_FMT_RGB24; 后编译无问题,但是跑代码时系统崩溃。
若是保持 ctx.v4l2format = V4L2_PIX_FMT_NV12; 则代码运行无问题。

我现在将 源图像解码为V4L2_PIX_FMT_NV12形式,然后进行推理,会出现以下问题:
E RKNNAPI: rknn_input_set,  inputs[0].buf wrong, buf = 0x7f40001550, size = 173056 (min_size = 519168)!
rknn_input_set fail! ret=-5


后续我将size*3,代码不会出现这个问题,但是会报出段错误。


我想请教下版主,官方提供的RKNN模型是否只支持RGB或BGR格式的图片。
如果我想推理NV12图片,怎么操作比较合适。





回复

使用道具 举报

jefferyzhang

版主

积分
13574
沙发
发表于 2020-10-9 12:43:11 | 只看该作者
看你模型,跟NPU和rknn无关。我们只是计算单元,谁知道你数据表达的是什么意思
回复

使用道具 举报

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

本版积分规则

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


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