Toybrick

标题: 【RKNN-toolkit】.rknn模型输入输出通道顺序为whcn或cwhn的问题 [打印本页]

作者: ZhuXinpei 时间: 2021-12-8 19:20
标题: 【RKNN-toolkit】.rknn模型输入输出通道顺序为whcn或cwhn的问题
芯片平台：RV1126
RKNN-toolkit版本：1.6.0

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

[attach]2186[/attach][attach]2187[/attach]

作者: jefferyzhang 时间: 2021-12-9 12:25
caffe本身就是whcn，而tensorflow和pytorch本身就是nhwc。这跟rknn无关。
请自行研究caffe和tf的区别。。。

欢迎光临 Toybrick (https://t.rock-chips.com/)