神经网络中的 HWC→CHW 格式转换

在神经网络中，HWC 和 CHW 是两种常见的数据格式，

HWC 代表高度（Height）、宽度（Width）、通道（Channels）

CHW 则是通道在前，代表通道（Channels）、高度（Height）、宽度（Width）。

HWC→CHW 的转换在神经网络中具有重要意义，以下是相关介绍：

• 转换原因
  • 框架设计要求 ：许多深度学习框架，如 PyTorch，其卷积操作（如nn.Conv2d）要求输入张量的维度顺序为 CHW。如果输入数据是 HWC 格式，会引发维度不匹配错误。
  • 计算效率与内存优化：CHW 格式在内存中按通道连续存储，更符合卷积操作中 "跨通道特征融合" 的计算模式，减少内存跳跃访问带来的延迟。现代 GPU 和深度学习框架（如 cuDNN）针对 NCHW（批量大小、通道、高度、宽度）格式进行了深度优化，而 CHW 是 NCHW 在单张图像情况下的表现形式，所以将 HWC 转换为 CHW 有利于提高计算效率。
  • 与预训练模型兼容：大多数预训练模型（如 ResNet、VGG）的权重是基于 NCHW 格式数据训练的，输入数据需为 CHW 或 NCHW 格式，否则模型会无法正确计算。
• 转换方法：可以使用维度置换函数来实现 HWC 到 CHW 的转换。

例如，在 Python 中使用 NumPy 库时，可以通过transpose函数进行转换，

如image_chw = image_hwc.transpose(2, 0, 1)；

在 PyTorch 中，可以使用permute函数，

如image_chw = image_hwc.permute(2, 0, 1)。