gpt4 book ai didi

image - 将 Pytorch 的图像格式从 NHWC 更改为 NCHW

转载 作者:行者123 更新时间:2023-12-02 02:01:15 26 4
gpt4 key购买 nike

在 pytorch 中,我们需要 NCHW 格式的图像,但我的图像是 NHWC

将此图像输入 CNN 的程序是什么? (我发现 this solution 建议使用“permute”,但我应该在哪里以及如何使用它?)

最佳答案

使用torch.Tensor.permute() :

 x = x.permute(0, 3, 1, 2) # from NHWC to NCHW

关于image - 将 Pytorch 的图像格式从 NHWC 更改为 NCHW,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/51881481/

26 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com