利用openCV+python进行HSV颜色识别,并结合滑动条动态改变目标颜色

HSV颜色空间更容易表示一个特定颜色,通过opencv读取的图片为BGR颜色空间,我们利用cv2.cvtColor()可以轻松实现颜色空间的转变。

 

在函数之前,我们先大致了解一下HSV基本颜色分量范围(通过实验得到的模糊范围,实际操作中我们可以据此做出适当调整)。

利用openCV+python进行HSV颜色识别,并结合滑动条动态改变目标颜色_第1张图片

图片来自:https://blog.csdn.net/taily_duan/article/details/51506776

 

函数学习

1. cv2.inRange(src, lowerb, upperb)

 

作用:

官方解释就是检查数组元素是否位于另外两个数组的元素之间,简言之就是找出指定范围内的颜色,并返回一个二值图像,指定范围内的图像显示为白色,相反为黑色。(注:cv2.inRange()函数对应的颜色空间为HSV)

 

参数:

  • src:输入图像
  • lowerb:颜色范围下边界(lower boundary)
  • upperb:颜色范围上边界(upper boundary)

 

e.g.

import cv2
import numpy as np
#载入原图
img_original=cv2.imread('E:\ShannonT\\notebook workspace\\images\\qicai.png')
#转变为HSV颜色空间
img_hsv=cv2.cvtColor(img_original,cv2.COLOR_BGR2HSV)
#返回黄色区域的二值图像
img_yellow=cv2.inRange(img_original,(27,160,215),(83,255,255))
cv2.imshow('img_original',img_original)
cv2.imshow('img_target',img_target)
cv2.waitKey()
cv2.destroyAllWindows() 

程序输出如下:

利用openCV+python进行HSV颜色识别,并结合滑动条动态改变目标颜色_第2张图片

2. cv2.bitwise_and(src1,src2[,mask])

 

作用:

将图像1和图像2进行按位与操作(bitwise:按位)

 

参数:

  • src1:输入图像1
  • src2:输入图像2
  • mask:可选参数,掩模(可以理解为一个窗口,作用为实现指定区域内的图像操作)

 

e.g.

import cv2
import numpy as np
#载入原图
img_original=cv2.imread('E:\ShannonT\\notebook workspace\\images\\qicai.png')
#转变为HSV颜色空间
img_hsv=cv2.cvtColor(img_original,cv2.COLOR_BGR2HSV)
#返回黄色区域的二值图像
img_yellow=cv2.inRange(img_original,(27,160,215),(83,255,255))
##输入图像与输入图像在掩模条件下按位与,得到掩模范围内的原图像
img_specifiedColor=cv2.bitwise_and(img_original,img_original,mask=img_yellow)
cv2.imshow('img_yellow',img_specifiedColor)
cv2.waitKey()
cv2.destroyAllWindows()

程序输出如下:

利用openCV+python进行HSV颜色识别,并结合滑动条动态改变目标颜色_第3张图片

 

下面通过一个具体的案例实现图像内黄色信息的识别,并加入滑动条的功能,可以让用户更直观地体验HSV颜色空间。

import cv2
import numpy as np
#定义窗口名称
winName='Colors of the rainbow'
#定义滑动条回调函数,此处pass用作占位语句保持程序结构的完整性
def nothing(x):
    pass
img_original=cv2.imread('E:\ShannonT\\notebook workspace\\images\\qicai.png')
#颜色空间的转换
img_hsv=cv2.cvtColor(img_original,cv2.COLOR_BGR2HSV)
#新建窗口
cv2.namedWindow(winName)
#新建6个滑动条,表示颜色范围的上下边界,这里滑动条的初始化位置即为黄色的颜色范围
cv2.createTrackbar('LowerbH',winName,27,255,nothing)
cv2.createTrackbar('LowerbS',winName,160,255,nothing)
cv2.createTrackbar('LowerbV',winName,215,255,nothing)
cv2.createTrackbar('UpperbH',winName,83,255,nothing)
cv2.createTrackbar('UpperbS',winName,255,255,nothing)
cv2.createTrackbar('UpperbV',winName,255,255,nothing)
while(1):
    #函数cv2.getTrackbarPos()范围当前滑块对应的值
    lowerbH=cv2.getTrackbarPos('LowerbH',winName)
    lowerbS=cv2.getTrackbarPos('LowerbS',winName)
    lowerbV=cv2.getTrackbarPos('LowerbV',winName)
    upperbH=cv2.getTrackbarPos('UpperbH',winName)
    upperbS=cv2.getTrackbarPos('UpperbS',winName)
    upperbV=cv2.getTrackbarPos('UpperbV',winName)
    #得到目标颜色的二值图像,用作cv2.bitwise_and()的掩模
    img_target=cv2.inRange(img_original,(lowerbH,lowerbS,lowerbV),(upperbH,upperbS,upperbV))
    #输入图像与输入图像在掩模条件下按位与,得到掩模范围内的原图像
    img_specifiedColor=cv2.bitwise_and(img_original,img_original,mask=img_target)
    cv2.imshow(winName,img_specifiedColor)
    if cv2.waitKey(1)==ord('q'):
        break
cv2.destroyAllWindows()

GIF效果如下:

 

#欢迎关注我的微信公众号

利用openCV+python进行HSV颜色识别,并结合滑动条动态改变目标颜色_第4张图片

你可能感兴趣的:(openCV,python)