修复 “Can't write images with one color channel” 错误的完整指南

在进行图像处理和保存时，我们可能会遇到以下错误：

1

ValueError: Can't write images with one color channel.

这是一个常见的问题，尤其是在使用 Python 的图像处理库（如 skimage 或 imageio）时。如果你也遇到了类似的错误，这篇博客将为你提供解决方法，并解释问题的原因。

背景

用户 @wern44 在运行其深度学习推理脚本时，尝试保存生成的图像，却触发了以下错误：

1

ValueError: Can't write images with one color channel.

错误发生在以下代码中：

1
2

io.imsave(os.path.join(result_path, im_name+".png"),
          (result*255).permute(1, 2, 0).cpu().data.numpy().astype(np.uint8))

这段代码试图将单通道图像保存为 PNG 文件，但 skimage.io.imsave 或 imageio 的底层实现不支持单通道图像的直接保存。

错误原因

该错误的核心问题在于 单通道图像的保存。许多图像处理库（如 skimage 或 imageio）在保存灰度图像时会出现问题，因为它们期望图像的形状为以下两种之一：

1. 多通道彩色图像：形状为 (height, width, 3)，每个像素点有 3 个通道（如 RGB）。
2. 单通道灰度图像：形状为 (height, width)。

导致错误的原因在于，传递给 imsave 的图像虽然是单通道，但形状却为 (height, width, 1)，即多了一个冗余的通道维度。这种形状无法被正确解析，因此抛出了 ValueError。

解决方法

针对该问题，推荐使用 OpenCV 替代 skimage 或 imageio 来保存图像。以下是修改后的代码：

1
2
3
4
5

import cv2

# 保存图像
cv2.imwrite(os.path.join(result_path, im_name + ".png"),
            (result * 255).permute(1, 2, 0).cpu().data.numpy().astype(np.uint8))

为什么 OpenCV 可行？

OpenCV 的 cv2.imwrite 方法能够处理形状为 (height, width) 或 (height, width, channels) 的图像：

• 如果图像是单通道灰度图（形状为 (height, width)），OpenCV 会自动将其存储为灰度图。
• 如果图像是多通道（如 RGB），OpenCV 会正确保存为彩色图像。

替代方案

如果你仍希望使用 io.imsave 或其他工具，可以在保存前将图像的形状调整为 (height, width)：

1
2
3
4
5
6
7
8

from skimage import io
import numpy as np

# 将单通道图像的形状从 (height, width, 1) 调整为 (height, width)
gray_image = (result * 255).permute(1, 2, 0).cpu().data.numpy().astype(np.uint8).squeeze()

# 保存图像
io.imsave(os.path.join(result_path, im_name + ".png"), gray_image)

np.squeeze() 方法会移除多余的维度 (height, width, 1) 中的最后一个通道，使图像变为 (height, width) 的灰度图。

总结

问题复现

如果你遇到以下错误：

1

ValueError: Can't write images with one color channel.

这是因为单通道图像的形状为 (height, width, 1)，而非 (height, width)，导致库无法正确处理。

快速修复

1. 使用 OpenCV：

1
2

import cv2
cv2.imwrite("output.png", single_channel_image)

2. 调整图像形状（如果继续使用 io.imsave）：

1
2
3

import numpy as np
gray_image = single_channel_image.squeeze()  # 从 (height, width, 1) 变为 (height, width)
io.imsave("output.png", gray_image)

选择更适合的工具

• OpenCV 是一个强大且灵活的计算机视觉库，建议在需要保存图像时优先选择它。
• skimage 和 imageio 虽然功能强大，但在某些情况下对输入数据的形状要求更严格。

参考内容

• 官方错误说明
• 讨论与解决方法示例

这篇文章希望能帮助你快速解决该问题，并对图像处理库的行为有更深入的理解！如果你有其他问题，欢迎留言交流！ 🎉