pytorch->onnx-rknn 输入channel是否等于3决定推理结果是否正确

kkkaaa 发表于 2020-5-8 14:23
这个问题挺困扰我的，虽然找到可以保持输出一致的方法，但是总觉得很奇怪

也不知道是不是我哪里做错了，不 ...

显示指定NHWC或者NCHW，输入的数据也要是这个格式。如果不是的话，可以可以用np.transpose接口去转换。