Toybrick

onnx模型转换为rknn模型后,input shape被改变

xxyyzz

新手上路

积分
33
楼主
发表于 2023-6-25 10:15:10    查看: 1396|回复: 1 | [复制链接]    打印 | 只看该作者
我有个yolov5的onnx的模型,input shape是NCHW格式的。我转换为rknn的格式模型后,在程序里读取模型的输入的时候,显示的是NHWC的了。为什么转换成rknn,会改变input shape的格式啊?

环境:
RK3588
rknn-toolkit2
回复

使用道具 举报

jefferyzhang

版主

积分
12953
沙发
发表于 2023-6-29 11:17:06 | 只看该作者
因为计算机图片格式都是NHWC的,默认都会转为NHWC部署使用,方便软件开发者喂数据。
回复

使用道具 举报

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

本版积分规则

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


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