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

[复制链接] · 发表于 2021-12-8 19:20:45

芯片平台：RV1126
RKNN-toolkit版本：1.6.0

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

只看该作者 · 发表于 2021-12-9 12:25:33

caffe本身就是whcn，而tensorflow和pytorch本身就是nhwc。这跟rknn无关。
请自行研究caffe和tf的区别。。。

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

本帖子中包含更多资源