如何用ImageMagick或OpenCV提取色调通道生成黑白图用于图像分割

现有思路风险提示 #

• 白色、灰色、黑色等无彩色的饱和度为0，此时色调无实际意义，几乎所有图像处理工具都会将无彩色的色调值返回为0。如果待分割的白鞋本身是纯白色无彩色纹理，和白色背景的色调值完全一致，仅靠色调通道无法完成分割，需要结合饱和度、亮度通道辅助判断。
• 色调为环形属性，0°和360°为同一值，直接线性映射为黑白灰度图时，会出现色调接近0°和360°的区域实际色调差异极小，但灰度值差异极大的问题，仅当你的背景色调严格为0、目标色调和0有明显差距时，该映射方式可用。

ImageMagick 实现方案 #

1. 提取色调映射黑白图 #

以下命令直接输出色调0对应黑色、色调360对应白色的灰度图：
convert 输入图像.jpg -colorspace HSV -channel 0 -separate 色调映射图.png

说明：ImageMagick的HSV色彩空间中第一个通道为色调通道，取值范围0-255正好对应色调0°-360°，分离后直接就是符合要求的黑白映射图。

2. 完整背景去除命令 #

如果目标白鞋存在彩色纹理，和背景色调差明显，可直接用以下命令生成带透明背景的分割结果：

convert 输入图像.jpg \
-colorspace HSV -channel 0 -separate \
-threshold 1% -negate \
\( 输入图像.jpg \) +swap \
-alpha off -compose copy_opacity -composite \
带透明背景结果.png

可调整threshold后的百分比数值优化分割精度，数值越大，越多低色调区域会被判定为背景。

如果白鞋为纯白色，需结合饱和度通道避免误分割，调整命令如下：

convert 输入图像.jpg \
-colorspace HSV \
\( -clone 0 -channel 0 -separate -threshold 1% \) \
\( -clone 0 -channel 1 -separate -threshold 5% \) \
-compose darken -composite -negate \
\( 输入图像.jpg \) +swap \
-alpha off -compose copy_opacity -composite \
带透明背景结果.png

可调整第二个threshold的百分比数值控制饱和度过滤精度，数值越高，对鞋身彩色纹理的要求越高。

OpenCV/Python 实现方案 #

import cv2
import numpy as np

# 读取原始图像
img = cv2.imread("输入图像.jpg")
# 转换为HSV色彩空间
hsv = cv2.cvtColor(img, cv2.COLOR_BGR2HSV)

# 生成色调映射黑白图
# OpenCV中H通道取值范围为0-179，对应0-360°，缩放至0-255实现需求映射
h_channel = hsv[:, :, 0]
hue_map = cv2.convertScaleAbs(h_channel, alpha=255/179)
cv2.imwrite("色调映射图.png", hue_map)

# 生成分割掩码
# 色调大于2的区域判定为目标
h_mask = cv2.inRange(h_channel, 2, 179)
# 饱和度大于10的区域判定为目标
s_channel = hsv[:, :, 1]
s_mask = cv2.inRange(s_channel, 10, 255)
# 合并两个掩码，覆盖纯白鞋身的情况
mask = cv2.bitwise_or(h_mask, s_mask)

# 生成带透明背景的结果
b, g, r = cv2.split(img)
result = cv2.merge((b, g, r, mask))
cv2.imwrite("带透明背景结果.png", result)

内容的提问来源于stack exchange，提问作者simone