Toybrick

标题: 模型推理，输入图像分辨率与模型分辨率不一致，rga报错 [打印本页]
作者: zhengsy    时间: 2023-2-7 09:12
标题: 模型推理，输入图像分辨率与模型分辨率不一致，rga报错
本帖最后由 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]。
报下面错误：
  1. Read ./model/09.jpg ...
  2. img width = 1080, img height = 1613
  3. Loading mode...
  4. sdk version: 1.4.0 (a10f100eb@2022-09-09T09:07:14) driver version: 0.8.2
  5. model input num: 1, output num: 3
  6.   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
  7.   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
  8.   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
  9.   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
  10. model is NHWC input fmt
  11. model input height=640, width=640, channel=3
  12. resize with RGA!
  13. rga_api version 1.8.0_[0]
  14. 243, check error! Unsupported function: src unsupport width stride 1080, rgb888 width stride should be 16 aligned!
复制代码更换代码运行的librga库任然报错，是版本不匹配导致的问题吗?如果不是需要怎样进行修改？请大佬指点。


作者: jefferyzhang    时间: 2023-2-7 15:40
rga是你自己调用的还是rknn内部调用的？
作者: zhengsy    时间: 2023-2-7 16:38
是yolov5物体检测代码中调用的用来进行图像缩放使用的
作者: jefferyzhang    时间: 2023-2-9 10:43
zhengsy 发表于 2023-2-7 16:38
是yolov5物体检测代码中调用的用来进行图像缩放使用的

代码谁写的



欢迎光临 Toybrick (https://t.rock-chips.com/) Powered by Discuz! X3.3