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

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

转载作者：行者123 更新时间：2023-12-02 02:01:15

26

4

在 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

文章推荐： java - 我是否需要发布到 Play 商店才能转换到 chromecast？

文章推荐： django 1.11 与 celery 4.0 和 djcelery 兼容性问题

文章推荐： java - 相当于python中的objectmapper

python - 当数据格式为 "NCHW"时，如何在tensorboard中显示图像？
在tensorflow中，使用函数tf.summary.image，我们可以在tensorboard中显示图像。 image( name, tensor, max_outputs=3, collect
tensorflow2.0 - 如何将保存的模型从 NCHW 转换为 NHWC？
我在 GPU 上训练了我的 NCHW 模型并保存了最佳状态。我现在想对 CPU 进行推断，它显然只支持 NHWC(我得到了一个错误提及)。我是否必须使用 NHWC 重新训练我的模型，或者有没有办法
tensorflow - 将张量从 NHWC 转置到 NCHW 有什么充分的理由吗？
经常在tensorflow代码中看到transpose的实现。我想知道为什么有人会想要将 NHWC 张量转置为 NCHW。请给我一个很好的例子及其背后的原因。最佳答案而不是引用文档。您应该阅读 C
image - 将 Pytorch 的图像格式从 NHWC 更改为 NCHW
在 pytorch 中，我们需要 NCHW 格式的图像，但我的图像是 NHWC。将此图像输入 CNN 的程序是什么？ (我发现 this solution 建议使用“permute”，但我应该在哪里
tensorflow - 在 TensorFlow 中在 NHWC 和 NCHW 之间转换
将张量从 NHWC 格式转换为 NCHW 格式(反之亦然)的最佳方法是什么？是否有专门执行此操作的操作，或者我需要使用拆分/连接类型操作的某种组合？最佳答案您需要做的就是将维度从 NHWC 排列
python - 具有 NCHW 格式的 tensorflow.nn.conv2d 中的过滤器形状
正在关注 Tensorflow's best practices for performance ，我正在使用 NCHW 数据格式，但我不确定要在 tensorflow.nn.conv2d 中使用的过
tensorflow - 与 TensorFlow/cuDNN 中的 NHWC 相比，NCHW 快多少？
The official TensorFlow performance guide指出: Most TensorFlow operations used by a CNN support both N
tensorflow - 将 tensorflow 数据格式从 nhwc( channel 在后)切换到 nchw( channel 在前)时出现问题
我目前正在尝试为 OCR 任务训练 CRNN(卷积循环神经网络)。我通常保持默认的数据格式，即 NHWC( channel 最后)。但是由于一些部署限制，我想使用 NCHW( channel 优先)数

首页

博学

6Ren·AI

商城

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