【发布时间】:2019-01-23 16:12:41
【问题描述】:
在 pytorch 中,我们需要 NCHW 格式的图像,但我的图像是 NHWC。
将此图像输入 CNN 的过程是什么? (我发现this solution 建议使用“permute”,但我应该在哪里以及如何使用它?)
【问题讨论】:
标签: image deep-learning permutation conv-neural-network pytorch
在 pytorch 中,我们需要 NCHW 格式的图像,但我的图像是 NHWC。
将此图像输入 CNN 的过程是什么? (我发现this solution 建议使用“permute”,但我应该在哪里以及如何使用它?)
【问题讨论】:
标签: image deep-learning permutation conv-neural-network pytorch
x = x.permute(0, 3, 1, 2) # from NHWC to NCHW
【讨论】: