37 lines
1.7 KiB
Python
37 lines
1.7 KiB
Python
import sys
|
||
sys.path.remove('/opt/ros/kinetic/lib/python2.7/dist-packages')
|
||
import cv2
|
||
import numpy as np
|
||
#定义窗口名称
|
||
winName='Colors of the rainbow'
|
||
#定义滑动条回调函数，此处pass用作占位语句保持程序结构的完整性
|
||
def nothing(x):
|
||
pass
|
||
img_original=cv2.imread('lane.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_hsv,(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() |