快捷導(dǎo)航

python+opencv處理顏色之將目標顏色轉(zhuǎn)換實例代碼

更新時間：2025年03月29日 09:54:16 作者：__基本操作__

OpenCV 是一個的跨平臺計算機視覺庫,可以運行在 Linux、Windows 和 Mac OS 操作系統(tǒng)上,這篇文章主要介紹了python+opencv處理顏色之將目標顏色轉(zhuǎn)換的相關(guān)資料,需要的朋友可以參考下

下面是代碼+ 效果 + 解釋

import cv2
import numpy as np

def dectfun(frame):
    # Convert the frame to HSV color space
    hsv = cv2.cvtColor(frame, cv2.COLOR_BGR2HSV)
    
    # Define the range for red color in HSV
    lower_red1 = np.array([0, 120, 70])
    upper_red1 = np.array([10, 255, 255])
    lower_red2 = np.array([170, 120, 70])
    upper_red2 = np.array([180, 255, 255])
    
    # Create a mask for red color
    mask1 = cv2.inRange(hsv, lower_red1, upper_red1)
    mask2 = cv2.inRange(hsv, lower_red2, upper_red2)
    mask = mask1 + mask2
    
    # Create an output image to show the detected red regions in green
    result = frame.copy()
    result[mask > 0] = [0, 255, 0]  # Change red regions to green
    
    return result

def main():
    # Open the webcam
    cap = cv2.VideoCapture(1)
    while True:
        # Read a frame
        ret, frame = cap.read()
        if not ret:
            break
        
        # Detect red regions and highlight them
        result = dectfun(frame)
        
        # Display the result
        cv2.imshow("Frame", result)
        
        if cv2.waitKey(1) & 0xFF == ord('q'):
            break
    
    # Release the webcam and close all windows
    cap.release()
    cv2.destroyAllWindows()

if __name__ == "__main__":
    main()

轉(zhuǎn)HSV：關(guān)于顏色總是要轉(zhuǎn)HSV的

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

掩膜

lower_red1 = np.array([0, 120, 70])
upper_red1 = np.array([10, 255, 255])
lower_red2 = np.array([170, 120, 70])
upper_red2 = np.array([180, 255, 255])

# Create a mask for red color
mask1 = cv2.inRange(hsv, lower_red1, upper_red1)
mask2 = cv2.inRange(hsv, lower_red2, upper_red2)
mask = mask1 + mask2

在這段代碼中，mask 是一個數(shù)組。具體來說，它是一個二值圖像（即一個包含0和255的數(shù)組），表示在HSV顏色空間中紅色區(qū)域的掩膜。在掩膜區(qū)域內(nèi)的是我們的目標lower_red1, upper_red1 標注是255 其余是0

再標注

    # Create an output image to show the detected red regions in green
    result = frame.copy()
    result[mask > 0] = [0, 255, 0]  # Change red regions to green
    
    return result

result = frame.copy()在Python中，變量通常是引用類型，如果你在函數(shù)中直接修改傳入的數(shù)組或?qū)ο螅鼈兊闹禃谰酶淖儭?/li>
result[mask > 0] = [0, 255, 0]這是個巧妙的句子，當mask的對應(yīng)位置上的值大于零。然后把result 對應(yīng)位置上的像素改成綠色