基于OpenCV与ObjC++提取图像特定颜色的技术问题求助

嘿，我明白你用ObjC++结合OpenCV做颜色提取时遇到了结果异常的情况——这种情况在颜色分割任务里太常见了，咱们一步步来排查和解决：

1. 优先切换到更适合的颜色空间 #

直接在RGB颜色空间做阈值分割很容易受光照、阴影影响，导致提取结果不干净。优先用HSV颜色空间，它把颜色的色调（H）、饱和度（S）、明度（V）分开，对光照变化的鲁棒性更强，更适合精准提取特定颜色。

在ObjC++里转换颜色空间的代码示例：

// 假设你已经把UIImage转换成了cv::Mat（注意OpenCV默认是BGR通道顺序）
cv::Mat bgrMat;
// 将BGR格式的Mat转成HSV
cv::Mat hsvMat;
cv::cvtColor(bgrMat, hsvMat, cv::COLOR_BGR2HSV);

这里要注意：UIImage的像素是RGB顺序，转成OpenCV的Mat时会自动变成BGR，所以用COLOR_BGR2HSV是正确的，别搞反通道顺序——这是很多人踩的隐形坑！

2. 精准调整颜色阈值范围 #

结果异常大概率是阈值没设准，尤其是红色这种在HSV里跨0度的特殊颜色：

• 红色区域提取：红色的H值范围是两段（[0, 10]和[160, 180]），S和V可以设[40, 255]和[40, 255]（可根据你的实际光照微调）
• 蓝色立方体提取：蓝色的H值范围大概是[90, 130]，S和V同样设[40, 255]和[40, 255]

对应ObjC++代码：

// 提取红色区域（合并两段阈值的mask）
cv::Mat redMask1, redMask2, redMask;
cv::inRange(hsvMat, cv::Scalar(0, 40, 40), cv::Scalar(10, 255, 255), redMask1);
cv::inRange(hsvMat, cv::Scalar(160, 40, 40), cv::Scalar(180, 255, 255), redMask2);
cv::bitwise_or(redMask1, redMask2, redMask);

// 提取蓝色立方体
cv::Mat blueMask;
cv::inRange(hsvMat, cv::Scalar(90, 40, 40), cv::Scalar(130, 255, 255), blueMask);

3. 加入后处理消除噪声 #

提取出来的mask可能会有很多小噪点或者空洞，用腐蚀（erode）和膨胀（dilate）组合的形态学操作来优化：

// 创建一个3x3的矩形结构元素
cv::Mat kernel = cv::getStructuringElement(cv::MORPH_RECT, cv::Size(3, 3));
// 开运算（先腐蚀再膨胀）：消除小噪点
cv::morphologyEx(redMask, redMask, cv::MORPH_OPEN, kernel);
// 闭运算（先膨胀再腐蚀）：填补mask里的小空洞
cv::morphologyEx(blueMask, blueMask, cv::MORPH_CLOSE, kernel);

4. 核对UIImage和Mat的转换逻辑 #

如果前面都没问题，那可能是图像转换时出了错。这里给你一个可靠的UIImage转cv::Mat的ObjC++实现：

+ (cv::Mat)cvMatFromUIImage:(UIImage *)image {
    CGColorSpaceRef colorSpace = CGImageGetColorSpace(image.CGImage);
    CGFloat cols = image.size.width;
    CGFloat rows = image.size.height;

    cv::Mat cvMat(rows, cols, CV_8UC4); // 8位4通道（RGBA）

    CGContextRef contextRef = CGBitmapContextCreate(cvMat.data, cols, rows,
                                                    8, cvMat.step[0], colorSpace,
                                                    kCGImageAlphaNoneSkipLast | kCGBitmapByteOrderDefault);

    CGContextDrawImage(contextRef, CGRectMake(0, 0, cols, rows), image.CGImage);
    CGContextRelease(contextRef);
    CGColorSpaceRelease(colorSpace);

    // 转成OpenCV默认的BGR格式
    cv::Mat bgrMat;
    cv::cvtColor(cvMat, bgrMat, cv::COLOR_RGBA2BGR);
    return bgrMat;
}

最后，把处理后的mask和原图结合，就能得到只保留目标颜色的图像：

cv::Mat result;
cv::bitwise_and(bgrMat, bgrMat, result, mask=redMask); // 提取红色区域
// 或者 cv::bitwise_and(bgrMat, bgrMat, result, mask=blueMask); 提取蓝色立方体

你可以先从调整HSV阈值入手，这是最容易见效的。如果还是有问题，可以先把mask可视化出来，判断是阈值不对还是噪声/转换逻辑的问题。

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