OpenCV如何检测帧中的特定颜色(inRange函数)

时间:2018-03-08 21:36:06

标签: python opencv machine-learning computer-vision opencv3.1

我可以使用下面的代码在框架中找到任何蓝色:

How To Detect Red Color In OpenCV Python?

但是我想修改代码以在我拥有的视频中查找非常特定的颜色。我从视频文件中读取一帧,然后将帧转换为HSV,然后在特定元素(包含我想要检测的颜色)上打印HSV值:

print("hsv[x][y]: {}".format(hsv[x][y]))

" HSV"是我使用cvtColor()将帧从BGR转换为HSV后得到的。 上面的打印命令给了我: hsv [x] [y]:[108 27 207]

然后我定义我的下HSV值和上HSV值,并将其传递给inRange()函数:

lowerHSV = np.array([107,26,206])

upperHSV = np.array([109,28,208])

maskHSV = cv2.inRange(hsv, lowerHSV, upperHSV)

我显示maskHSV,但它似乎并不能识别包含该颜色的项目。我尝试扩展较低的HSV和较高的HSV界限,但这似乎也无效。 我尝试使用BGR类似的东西,但似乎没有效果。

我试图识别的东西最好被描述为柠檬 - 酸橙运动饮料瓶......

任何建议都将不胜感激。

=============================================== ==

我正在运行的完整python代码如下所示,以及一些相关的图像......

import cv2
import numpy as np 
import time

video_capture = cv2.VideoCapture("conveyorBeltShort.wmv") 

xloc = 460

yloc = 60

dCounter = 0

while(1):

    dCounter += 1

    grabbed, frame = video_capture.read()

    if grabbed == False:
        break

    time.sleep(2)

    hsv = cv2.cvtColor(frame, cv2.COLOR_BGR2HSV)

    lowerBGR = np.array([200,190,180])
    upperBGR = np.array([210,200,190])

    # HSV [108 27 209] observation
    lowerHSV = np.array([105,24,206])
    upperHSV = np.array([111,30,212])

    maskBGR = cv2.inRange(frame, lowerBGR, upperBGR)
    maskHSV = cv2.inRange(hsv, lowerHSV, upperHSV)

    cv2.putText(hsv, "HSV:" + str(hsv[xloc][yloc]),(100,280), 
        cv2.FONT_HERSHEY_SIMPLEX, 3,(0,0,0),8,cv2.LINE_AA)
    cv2.putText(frame, "BGR:" + str(frame[xloc][yloc]),(100,480), 
        cv2.FONT_HERSHEY_SIMPLEX, 3,(0,0,0),8,cv2.LINE_AA)
    cv2.rectangle(frame, (0, 0), (xloc-1, yloc-1), (255, 0, 0), 2)
    cv2.rectangle(hsv, (0, 0), (xloc-1, yloc-1), (255, 0, 0), 2)

    cv2.imwrite("maskHSV-%d.jpg" % dCounter, maskHSV)
    cv2.imwrite("maskBGR-%d.jpg" % dCounter, maskBGR)
    cv2.imwrite("hsv-%d.jpg" % dCounter, hsv) 
    cv2.imwrite("frame-%d.jpg" % dCounter, frame)

    cv2.imshow('frame',frame)
    cv2.imshow('hsv',hsv)
    cv2.imshow('maskHSV',maskHSV)    
    cv2.imshow('maskBGR',maskBGR)

    if cv2.waitKey(1) & 0xFF == ord('q'):
        break

cv2.destroyAllWindows()
video_capture.release()

=============================================== ==

第一张图片是" frame-6.jpg"

第二张图片是" hsv-6.jpg"

第三张图片是" maskHSV-6.jpg"

第四张图片是" maskBGR-6.jpg"

maskHSV-6.jpg和maskBGR-6.jpg似乎没有在传送带上显示柠檬灰瓶。我相信我已正确设置了较低和较高的HSV / BGR限制......

frame-6.jpg

hsv-6.jpg

maskHSV-6.jpg

maskBGR-6.jpg

1 个答案:

答案 0 :(得分:0)

我只知道OpenCV的C ++,但根据this post你应该使用img [y] [x]或img [x,y]来访问像素值。

你的rgb值不正确。它应该是[96,160,165]