使用Python和OpenCV庫實現(xiàn)實時顏色識別系統(tǒng)
更新時間：2025年06月16日 11:13:12 作者：E-An居士
這篇文章主要介紹了使用Python和OpenCV庫實現(xiàn)的實時顏色識別系統(tǒng),這個系統(tǒng)能夠通過攝像頭捕捉視頻流,并在視頻中指定區(qū)域內識別主要顏色（紅、黃、綠、藍）,這種技術在機器人視覺、自動化檢測和交互式應用中有著廣泛的應用前景,需要的朋友可以參考下
一、引言
今天我將介紹一個使用Python和OpenCV庫實現(xiàn)的實時顏色識別系統(tǒng)。這個系統(tǒng)能夠通過攝像頭捕捉視頻流，并在視頻中指定區(qū)域內識別主要顏色（紅、黃、綠、藍）。這種技術在機器人視覺、自動化檢測和交互式應用中有著廣泛的應用前景。
二、系統(tǒng)概述
該系統(tǒng)主要包含以下幾個功能：
- 實時視頻捕捉
- 在視頻幀中劃定特定檢測區(qū)域
- 將檢測區(qū)域轉換為HSV色彩空間
- 分析區(qū)域內的色調(H)值
- 根據H值范圍判斷顏色類型
- 實時顯示結果
三、代碼解析
1. 導入庫
import cv2
我們只需要導入OpenCV庫，它提供了強大的計算機視覺功能，包括圖像處理和視頻捕捉。
2. 顏色識別函數(shù)
def get_color(img):
H = []
color_name = None
img = cv2.resize(img,(640,400),)
# 將彩色圖轉換為HSV
HSV = cv2.cvtColor(img,cv2.COLOR_BGR2HSV)
# 畫矩形框
cv2.rectangle(img,(280,180),(360,260),(0,255,0),2)
# 依次取出每行每列的H，S,V值放入容器中
for i in range(280,360):
for j in range(180,260): H.append(HSV[j,i][0])
# 分別計算出H,S,V的最大最小
H_min = min(H);H_max = max(H)
# 判斷顏色
if H_min >= 0 and H_max <= 10 or H_min >= 156 and H_max <= 180: color_name='red'
elif H_min >= 26 and H_max <= 34 : color_name='yellow'
elif H_min >= 35 and H_max <= 77 : color_name='green'
elif H_min >= 100 and H_max <= 124 : color_name='blue'
print(color_name)
return img,color_name
函數(shù)功能詳解：
圖像預處理：
- 首先將輸入圖像調整為固定尺寸(640×400)，確保處理一致性
色彩空間轉換：
- 將BGR格式轉換為HSV格式，HSV色彩空間更適合顏色識別
- H(色調)：表示顏色類型
- S(飽和度)：表示顏色的純度
- V(亮度)：表示顏色的明暗程度
檢測區(qū)域標記：
- 在圖像上繪制一個綠色矩形框(280,180)到(360,260)，標識檢測區(qū)域
數(shù)據采集：
- 遍歷檢測區(qū)域內每個像素點，收集所有H值
顏色判斷：
- 計算區(qū)域內H值的最小和最大值
- 根據H值范圍判斷顏色：
- 紅色：0-10或156-180
- 黃色：26-34
- 綠色：35-77
- 藍色：100-124
3. 主程序循環(huán)
cap = cv2.VideoCapture(0)
while 1:
_,frame=cap.read()
img,cal = get_color(frame)
cv2.imshow('',img)
if cv2.waitKey(1) == 27:
break
視頻捕捉初始化：
- 創(chuàng)建VideoCapture對象，參數(shù)0表示使用默認攝像頭
主循環(huán)：
- 不斷讀取視頻幀
- 調用get_color函數(shù)處理每一幀
- 顯示處理后的圖像
- 按ESC鍵(ASCII 27)退出程序
四、HSV色彩空間詳解
為什么選擇HSV而不是RGB進行顏色識別？
RGB的局限性：
- RGB三個通道都與亮度相關
- 對光照變化敏感
- 顏色判斷需要同時考慮三個通道
HSV的優(yōu)勢：
- 將顏色信息(H)與亮度(V)、飽和度(S)分離
- 對光照變化有一定魯棒性
- 顏色判斷主要依據H通道
五、顏色范圍設定
OpenCV中HSV的范圍：
- H: 0-180 (通常色彩空間為0-360，但OpenCV使用8位存儲，所以除以2)
- S: 0-255
- V: 0-255
常見顏色H值范圍：
- 紅色：0-10和170-180
- 橙色：11-25
- 黃色：26-34
- 綠色：35-77
- 藍色：100-124
- 紫色：125-155
可對比如下的顏色范圍圖
六、系統(tǒng)優(yōu)化建議
增加飽和度(S)和亮度(V)的過濾：
- 可以排除低飽和度(接近灰色)或低亮度(接近黑色)的區(qū)域
使用均值而非極值：
- 當前使用min/max容易受噪聲影響，可改為計算平均值
擴大檢測區(qū)域：
- 當前區(qū)域較小(80×80像素)，可以適當擴大
添加更多顏色識別：
- 如橙色、紫色等
優(yōu)化性能：
- 當前雙重循環(huán)效率不高，可以使用NumPy進行向量化操作
七、完整代碼
import cv2
def get_color(img):
H = []
color_name = None
img = cv2.resize(img,(640,400),)
# 將彩色圖轉換為HSV
HSV = cv2.cvtColor(img,cv2.COLOR_BGR2HSV)
# 畫矩形框
cv2.rectangle(img,(280,180),(360,260),(0,255,0),2)
# 依次取出每行每列的H，S,V值放入容器中
for i in range(280,360):
for j in range(180,260): H.append(HSV[j,i][0])
# 分別計算出H,S,V的最大最小
H_min = min(H);H_max = max(H)
# 判斷顏色
if H_min >= 0 and H_max <= 10 or H_min >= 156 and H_max <= 180: color_name='red'
elif H_min >= 26 and H_max <= 34 : color_name='yellow'
elif H_min >= 35 and H_max <= 77 : color_name='green'
elif H_min >= 100 and H_max <= 124 : color_name='blue'
print(color_name)
return img,color_name
cap = cv2.VideoCapture(0)
while 1:
_,frame=cap.read()
img,cal = get_color(frame)
cv2.imshow('',img)
if cv2.waitKey(1) == 27:
break
八、總結
本文介紹了一個基于OpenCV的實時顏色識別系統(tǒng)，通過HSV色彩空間轉換和色調范圍判斷實現(xiàn)了基本顏色識別功能。該系統(tǒng)可以作為更復雜計算機視覺項目的基礎，通過進一步優(yōu)化和擴展，能夠滿足各種實際應用需求。
以上就是使用Python和OpenCV庫實現(xiàn)實時顏色識別系統(tǒng)的詳細內容，更多關于Python OpenCV實時顏色識別的資料請關注腳本之家其它相關文章！
相關文章
pycharm 在windows上編輯代碼用linux執(zhí)行配置的方法
今天小編就為大家分享一篇pycharm 在windows上編輯代碼用linux執(zhí)行配置的方法，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2018-10-10
這篇文章主要介紹了如何在sublime編輯器中安裝python,文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值,需要的朋友可以參考下2020-05-05
利用Python?Difflib庫強大的文字比較功能快速輕松查重
這篇文章主要介紹了利用Python?Difflib庫強大的文字比較功能快速輕松查重實例詳解,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進步,早日升職加薪2024-01-01
這篇文章主要為大家詳細介紹了Python調用攝像頭拍攝數(shù)據集，具有一定的參考價值，感興趣的小伙伴們可以參考一下2019-06-06
今天小編就為大家分享一篇python直接獲取API傳遞回來的參數(shù)方法，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2018-12-12
這篇文章主要介紹了python實現(xiàn)人工智能Ai摳圖功能,本文通過實例代碼給大家介紹的非常詳細，具有一定的參考借鑒價值，需要的朋友可以參考下2019-09-09
這篇文章主要介紹了python協(xié)程用法,實例分析Python中協(xié)議的概念、功能及使用方法,需要的朋友可以參考下2015-06-06
Python獲取當前腳本文件夾(Script)的絕對路徑方法代碼
在本篇文章中小編給各位整理了關于Python獲取當前腳本文件夾(Script)的絕對路徑實例代碼內容，有需要的朋友們學習下。2019-08-08
本篇文章主要介紹了Python內置函數(shù)delattr的具體用法，小編覺得挺不錯的，現(xiàn)在分享給大家，也給大家做個參考。一起跟隨小編過來看看吧2017-11-11
詳解Numpy中的數(shù)組拼接、合并操作（concatenate, append, stack, hstack, vstac
這篇文章主要介紹了詳解Numpy中的數(shù)組拼接、合并操作（concatenate, append, stack, hstack, vstack, r_, c_等），具有一定的參考價值，感興趣的小伙伴們可以參考一下2019-05-05