模型推理，输入图像分辨率与模型分辨率不一致，rga报错

[复制链接] · 发表于 2023-2-7 09:12:57

本帖最后由 zhengsy 于 2023-2-7 09:25 编辑

在yoloyv5模型推理时，输入图像分辨率与模型输入分辨率调用rga进行resize报错。
我输入模型的分辨率为640*640，传入的推理图像分辨率要比640*640大，在模型推理时代码中使用RGA 进行resize。板子是rk3588的板子，使用指令查询系统安装的librga版本是rga_api version 1.8.1_[3]，代码运行打印版本为rga_api version 1.8.0_[0]。
报下面错误：

Read ./model/09.jpg ...

img width = 1080, img height = 1613

Loading mode...

sdk version: 1.4.0 (a10f100eb@2022-09-09T09:07:14) driver version: 0.8.2

model input num: 1, output num: 3

  index=0, name=data, n_dims=4, dims=[1, 640, 640, 3], n_elems=1228800, size=1228800, fmt=NHWC, type=INT8, qnt_type=AFFINE, zp=-128, scale=0.003922

  index=0, name=output, n_dims=4, dims=[1, 33, 80, 80], n_elems=211200, size=211200, fmt=NCHW, type=INT8, qnt_type=AFFINE, zp=64, scale=0.157604

  index=1, name=379, n_dims=4, dims=[1, 33, 40, 40], n_elems=52800, size=52800, fmt=NCHW, type=INT8, qnt_type=AFFINE, zp=60, scale=0.143637

  index=2, name=380, n_dims=4, dims=[1, 33, 20, 20], n_elems=13200, size=13200, fmt=NCHW, type=INT8, qnt_type=AFFINE, zp=31, scale=0.111892

model is NHWC input fmt

model input height=640, width=640, channel=3

resize with RGA!

rga_api version 1.8.0_[0]

243, check error! Unsupported function: src unsupport width stride 1080, rgb888 width stride should be 16 aligned!
复制代码

更换代码运行的librga库任然报错，是版本不匹配导致的问题吗?如果不是需要怎样进行修改？请大佬指点。

只看该作者 · 发表于 2023-2-7 15:40:22

rga是你自己调用的还是rknn内部调用的？

只看该作者 · 发表于 2023-2-7 16:38:04

是yolov5物体检测代码中调用的用来进行图像缩放使用的

只看该作者 · 发表于 2023-2-9 10:43:12

zhengsy 发表于 2023-2-7 16:38
是yolov5物体检测代码中调用的用来进行图像缩放使用的

代码谁写的