让我们加载同一个立方体的2个图像。 它将默认以BGR格式加载。 我们可以使用OpenCV函数cvtColor()在不同的颜色空间之间进行转换,如后面所示。
bright = cv::imread('cube1.jpg')
dark = cv::imread('cube8.jpg')
第一张照片是在阳光明媚的室外环境下拍摄的,第二张照片是在正常照明条件下拍摄的。
RGB色彩空间具有以下属性
让我们把这两幅图像分解成它们的R,G和B分量,并观察它们以获得更多的色彩空间。
如果你看蓝色的通道,可以看到在室内照明条件下,第二张图像中的蓝色和白色的部分看起来相似,但是在第一张图像中有明显的差别。 这种不均匀性使得在这个色彩空间中基于颜色的分割非常困难。 此外,两个图像的值之间存在总体差异。 下面我们总结了与RGB颜色空间相关的固有问题:
Lab色彩空间有三个组件。
1、L - 亮度(强度)。
2、a-从绿色到品红色的一种颜色成分。
3、b-颜色分量从蓝色到黄色。
Lab色彩空间与RGB色彩空间大不相同。 在RGB色彩空间中,色彩信息被分成三个通道,但相同的三个通道也编码亮度信息。 另一方面,在Lab色彩空间中,L通道独立于色彩信息,仅对亮度进行编码。 另外两个通道编码颜色。
它具有以下属性。
让我们看看Lab色彩空间中的两个图像分成三个通道。
cv::cvtColor(bright, brightLAB, cv::COLOR_BGR2LAB);
cv::cvtColor(dark, darkLAB, cv::COLOR_BGR2LAB);
YCrCb颜色空间是从RGB颜色空间导出的,并具有以下三个组件。
1、Y - 伽马校正后从RGB获得的亮度或亮度(Luma )分量。
2、Cr = R - Y(的红色成分距离Luma有多远)。
3、Cb = B - Y(蓝色分量距离Luma的有多远)。
此颜色空间具有以下属性。
下面显示了YCrCb色彩空间中分成两个通道的两幅图像
cv::cvtColor(bright, brightYCB, cv::COLOR_BGR2YCrCb);
cv::cvtColor(dark, darkYCB, cv::COLOR_BGR2YCrCb);
HSV色彩空间有以下三个组成部分
1、H - 色调(主波长)。
2、S - 饱和度(纯度/颜色的阴影)。
3、V值(强度)。
我们列举一些属性。
两幅图像的H,S和V分量如下所示。
cv::cvtColor(bright, brightHSV, cv::COLOR_BGR2HSV);
cv::cvtColor(dark, darkHSV, cv::COLOR_BGR2HSV);
现在我们已经对不同的色彩空间有了一些了解,我们先尝试用它们来检测立方体中的绿色。
第1步:获取特定颜色的颜色值
为每个色彩空间查找绿色值的近似范围。 为此,我创建了一个交互式GUI,您可以通过将鼠标悬停在图像上来检查每个像素的所有颜色空间的值,如下所示:
步骤2:应用阈值进行分割
从图像中提取具有与绿色像素值接近的所有像素。 我们可以为每个色彩空间取+/- 40的范围,并检查结果是怎样的。 我们将使用inRange中的opencv函数查找绿色像素的遮罩,然后使用bitwise_and操作使用遮罩从图像中获取绿色像素。
另外请注意,为了将一个像素转换为另一个色彩空间,我们首先需要将一维数组转换为三维数组。
//C++ code
cv::Vec3b bgrPixel(40, 158, 16);
// Create Mat object from vector since cvtColor accepts a Mat object
Mat3b bgr (bgrPixel);
//Convert pixel values to other color spaces.
Mat3b hsv,ycb,lab;
cvtColor(bgr, ycb, COLOR_BGR2YCrCb);
cvtColor(bgr, hsv, COLOR_BGR2HSV);
cvtColor(bgr, lab, COLOR_BGR2Lab);
//Get back the vector from Mat
Vec3b hsvPixel(hsv.at
Vec3b ycbPixel(ycb.at
Vec3b labPixel(lab.at
int thresh = 40;
cv::Scalar minBGR = cv::Scalar(bgrPixel.val[0] - thresh, bgrPixel.val[1] - thresh, bgrPixel.val[2] - thresh)
cv::Scalar maxBGR = cv::Scalar(bgrPixel.val[0] + thresh, bgrPixel.val[1] + thresh, bgrPixel.val[2] + thresh)
cv::Mat maskBGR, resultBGR;
cv::inRange(bright, minBGR, maxBGR, maskBGR);
cv::bitwise_and(bright, bright, resultBGR, maskBGR);
cv::Scalar minHSV = cv::Scalar(hsvPixel.val[0] - thresh, hsvPixel.val[1] - thresh, hsvPixel.val[2] - thresh)
cv::Scalar maxHSV = cv::Scalar(hsvPixel.val[0] + thresh, hsvPixel.val[1] + thresh, hsvPixel.val[2] + thresh)
cv::Mat maskHSV, resultHSV;
cv::inRange(brightHSV, minHSV, maxHSV, maskHSV);
cv::bitwise_and(brightHSV, brightHSV, resultHSV, maskHSV);
cv::Scalar minYCB = cv::Scalar(ycbPixel.val[0] - thresh, ycbPixel.val[1] - thresh, ycbPixel.val[2] - thresh)
cv::Scalar maxYCB = cv::Scalar(ycbPixel.val[0] + thresh, ycbPixel.val[1] + thresh, ycbPixel.val[2] + thresh)
cv::Mat maskYCB, resultYCB;
cv::inRange(brightYCB, minYCB, maxYCB, maskYCB);
cv::bitwise_and(brightYCB, brightYCB, resultYCB, maskYCB);
cv::Scalar minLAB = cv::Scalar(labPixel.val[0] - thresh, labPixel.val[1] - thresh, labPixel.val[2] - thresh)
cv::Scalar maxLAB = cv::Scalar(labPixel.val[0] + thresh, labPixel.val[1] + thresh, labPixel.val[2] + thresh)
cv::Mat maskLAB, resultLAB;
cv::inRange(brightLAB, minLAB, maxLAB, maskLAB);
cv::bitwise_and(brightLAB, brightLAB, resultLAB, maskLAB);
cv2::imshow("Result BGR", resultBGR)
cv2::imshow("Result HSV", resultHSV)
cv2::imshow("Result YCB", resultYCB)
cv2::imshow("Output LAB", resultLAB)
所以,RGB和LAB似乎足以检测颜色,我们不需要考虑太多。 让我们看看更多的结果。
所以,相同的门槛不适用于黑暗的形象。 做相同的实验来检测黄色给出以下结果。
但为什么结果如此糟糕呢? 这是因为我们猜测了40的门槛。 我做了另一个交互式演示,在这里你可以玩这些值,并试图找到一个适用于所有图像的演示。 看看截图。 但是,那么会出现另一个图像出现的情况,并且不能再次工作。 我们不能盲目地通过试错来取得一些门槛。 我们没有这样做,而是利用色彩空间的力量。
我在变化的照明条件下收集了10个立方体的图像,并分别裁剪每种颜色,以获得6种不同颜色的6个数据集。 你可以看到多少变化的颜色进行视觉。
检查不同颜色空间中特定颜色的分布,如蓝色或黄色。 密度图或二维直方图给出了一个给定颜色值的变化的想法。 例如,理想情况下,蓝色图像的蓝色通道应始终具有255的值。但实际上,它分布在0到255之间。
我只显示BGR色彩空间的代码。 你需要为所有的颜色空间。
可以看出,在相似的照明条件下,所有的地块都非常紧凑。 有一点要注意的是:
当照明变化很大时,我们可以看到:
在最后一节中,我将通过从密度图中获取阈值来显示检测蓝色和黄色片段的结果,并以与我们在第二部分中所做的相同的方式将其应用于各个色彩空间。 当我们在HSV,YCrCb和LAB色彩空间工作时,我们不必担心Intensity分量。 我们只需要指定颜色分量的阈值。 图中显示了我用来生成结果的值。
P.S:如果你对解决魔方有兴趣,可以参考这个循序渐进的指南。