Toybrick

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

zhengsy

新手上路

积分
18
楼主
发表于 2023-2-7 09:12:57    查看: 2128|回复: 3 | [复制链接]    打印 | 只看该作者
本帖最后由 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

版主

积分
12952
沙发
发表于 2023-2-7 15:40:22 | 只看该作者
rga是你自己调用的还是rknn内部调用的?
回复

使用道具 举报

zhengsy

新手上路

积分
18
板凳
 楼主| 发表于 2023-2-7 16:38:04 | 只看该作者
是yolov5物体检测代码中调用的用来进行图像缩放使用的
回复

使用道具 举报

jefferyzhang

版主

积分
12952
地板
发表于 2023-2-9 10:43:12 | 只看该作者
zhengsy 发表于 2023-2-7 16:38
是yolov5物体检测代码中调用的用来进行图像缩放使用的

代码谁写的
回复

使用道具 举报

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

本版积分规则

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


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