在 pytorch 中，我们需要 NCHW 格式的图像，但我的图像是 NHWC。
将此图像输入 CNN 的程序是什么？ (我发现 this solution 建议使用“permute”，但我应该在哪里以及如何使用它？)
最佳答案
x = x.permute(0, 3, 1, 2) # from NHWC to NCHW
关于image - 将 Pytorch 的图像格式从 NHWC 更改为 NCHW，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/51881481/