Toybrick

【RKNN-toolkit】.rknn模型输入输出通道顺序为whcn或cwhn的问题

ZhuXinpei

新手上路

积分
13
楼主
发表于 2021-12-8 19:20:45    查看: 3402|回复: 1 | [复制链接]    打印 | 只看该作者
芯片平台:RV1126
RKNN-toolkit版本:1.6.0

RK的工程师和开发者们您好:
我在使用RKNN-toolkit将模型转为.rknn格式后,通过Netron打开模型文件发现模型输入输出的通道顺序有些混乱。在我的实验结果中,从Caffe转出的模型输入输出格式都为whcn,从tensorflow(.pb)转出的模型输入输出格式都为cwhn(不确定)。我未能从官方文档中查找到有关可以控制通道顺序的接口。想请教如下问题:
1、可以确定从Netron中看到的通道顺序就是运行模型文件后输出数据的顺序吗?
2、默认从tensorflow(.pb)转出的模型的输出格式为cwhn吗?有接口可以改变通道顺序吗?



本帖子中包含更多资源

您需要 登录 才可以下载或查看,没有帐号?立即注册

x
回复

使用道具 举报

jefferyzhang

版主

积分
13572
沙发
发表于 2021-12-9 12:25:33 | 只看该作者
caffe本身就是whcn,而tensorflow和pytorch本身就是nhwc。这跟rknn无关。
请自行研究caffe和tf的区别。。。
回复

使用道具 举报

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

本版积分规则

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


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