RKNN能否推理NV12图片

[复制链接] · 发表于 2020-10-9 12:04:04

我本想根据官方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图片，怎么操作比较合适。

只看该作者 · 发表于 2020-10-9 12:43:11

看你模型，跟NPU和rknn无关。我们只是计算单元，谁知道你数据表达的是什么意思