如何在OpenCV中匹配指定RGB颜色并生成掩码？

问题描述 #

我想在OpenCV中匹配特定RGB颜色（比如纯绿色0x00, 0xff, 0x00）并生成对应区域的掩码，但使用cv::compare()函数时，它会独立处理三个通道，无法实现预期效果。测试代码如下：

cv::Mat testImg = cv::imread("test.png", cv::IMREAD_COLOR);
cv::Mat mask(testImg.size().height, testImg.size().width, CV_8U, cv::Scalar(0));
cv::compare(testImg, cv::Vec3b(0x00, 0xff, 0x00), mask, cv::CMP_EQ);
imwrite("out.png", mask);

预期匹配测试图第一行的绿色方块，但输出的掩码是按通道匹配0值（忽略0xff）得到的非布尔型反转颜色掩码，不符合需求。我可以拆分通道多次比较再合并，但想寻求更优方案：能否将3通道图像转为24位单通道一次性比较？或者是否应该使用inRange()？

解决方案 #

1. 优先使用cv::inRange()函数 #

这是OpenCV中匹配特定颜色最直接高效的方式，它能同时对多通道图像的每个像素进行范围判断，只需将上下限设为同一个颜色值，就能精准匹配单一颜色。示例代码：

cv::Mat testImg = cv::imread("test.png", cv::IMREAD_COLOR);
cv::Mat mask;
// 注意：OpenCV默认读取的彩色图像是*BGR格式*，所以这里要对应调整颜色顺序
cv::Scalar lower(0x00, 0xff, 0x00); // BGR顺序的纯绿色
cv::Scalar upper(0x00, 0xff, 0x00);
cv::inRange(testImg, lower, upper, mask);
cv::imwrite("out.png", mask);

生成的掩码中，匹配到绿色的区域为255（白色），未匹配区域为0（黑色），完全符合预期。

2. 不推荐转24位单通道的方案 #

理论上可以通过手动重组通道将3通道BGR图像转为24位单通道，但这种方式存在明显弊端：

• OpenCV对24位单通道格式的支持有限，多数内置函数无法直接处理
• 需要手动完成像素位拼接与比较，代码冗余且效率不如inRange()
  因此不建议采用该方案。

3. 拆分通道合并的备选方案 #

如果不想使用inRange()，也可以拆分通道后分别比较，再通过位运算合并结果，示例代码：

cv::Mat testImg = cv::imread("test.png", cv::IMREAD_COLOR);
std::vector<cv::Mat> channels;
cv::split(testImg, channels);
cv::Mat b_eq, g_eq, r_eq;
cv::compare(channels[0], 0x00, b_eq, cv::CMP_EQ); // 蓝通道等于0
cv::compare(channels[1], 0xff, g_eq, cv::CMP_EQ); // 绿通道等于255
cv::compare(channels[2], 0x00, r_eq, cv::CMP_EQ); // 红通道等于0
cv::Mat mask;
cv::bitwise_and(b_eq, g_eq, mask);
cv::bitwise_and(mask, r_eq, mask);
cv::imwrite("out.png", mask);

但该方案需要多次拆分、比较和位运算，简洁性与效率均不如inRange()。

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