Python opencv 根据HSV实现简单的颜色分类

最新推荐文章于 2024-02-05 02:04:58 发布

乐亦亦乐

最新推荐文章于 2024-02-05 02:04:58 发布

阅读量3.7k

点赞数

分类专栏： python OpenCV 文章标签： python

本文链接：https://blog.csdn.net/qq_41251963/article/details/123934945

版权

python OpenCV 专栏收录该内容

23 篇文章 2 订阅

订阅专栏

该博客通过OpenCV库展示了如何使用HSV色彩空间进行颜色选择和跟踪。首先创建了一个HSV颜色选择器窗口，用户可以通过滑块调整H、S、V值来选择颜色。然后，从摄像头捕获视频帧，将BGR图像转换为HSV，并在画面中心检测像素值，根据HSV值判断颜色类别，如红色、橙色、黄色等。最后，在帧上显示识别的颜色并画出中心点。

摘要由CSDN通过智能技术生成

# hsv_color_picker
"""
import cv2
import numpy as np

def nothing(x):
    pass

cv2.namedWindow("frame")
cv2.createTrackbar("H","frame",0,179,nothing)
cv2.createTrackbar("S","frame",255,255,nothing)
cv2.createTrackbar("V","frame",255,255,nothing)

img_hsv=np.zeros((250,500,3),np.uint8)

while True:
    h=cv2.getTrackbarPos("H","frame")
    s=cv2.getTrackbarPos("S","frame")
    v=cv2.getTrackbarPos("V","frame")

    img_hsv[:]=(h,s,v)
    img_bgr=cv2.cvtColor(img_hsv,cv2.COLOR_HSV2BGR)

    cv2.imshow("frame",img_bgr)
    key=cv2.waitKey(1)
    if key==27:
        break
cv2.destroyWindow()
"""

import cv2

cap=cv2.VideoCapture(1)
cap.set(cv2.CAP_PROP_FRAME_WIDTH,1280)
cap.set(cv2.CAP_PROP_FRAME_HEIGHT,720)


while True:
    _,frame=cap.read()
    hsv_frame=cv2.cvtColor(frame,cv2.COLOR_BGR2HSV)
    height,width,_=frame.shape

    cy,cx=height//2,width//2

    # 获取画面中心点像素值
    pixel_center=hsv_frame[cy,cx]
    hue_value=pixel_center[0]

    color="Undefined"
    if 0<hue_value<11 or 156<hue_value<180:
        color="RED"
    elif 11<hue_value<25:
        color="ORANGE"
    elif 26<hue_value<34:
        color="YELLOW"
    elif 35<hue_value<77:
        color="GREEN"
    elif 78<hue_value<99:
        color="Qing"
    elif 100<hue_value<124:
        color="BLUE"
    elif 125<hue_value<155:
        color="purple"
    # elif hue_value<170:
    #     color="PINK"
    else:
        pass
        # color="RED"


    pixel_center_bgr=frame[cx,cy]
    b,g,r=[int(i) for i in pixel_center_bgr]

    cv2.putText(frame,color,(10,70),0,1,(255,255,255),2)
    cv2.circle(frame,(cx,cy),8,(255,255,255),3)

    cv2.imshow("Frame",frame)
    key=cv2.waitKey(1)
    if key==27:
        break

cap.release()
cv2.destroyWindow()