tracking multiple objects by color OpenCV 2.x

后端 未结 1 536
余生分开走
余生分开走 2021-02-06 15:58

Currently i am trying to track multiple objects by color. I\'ve based on documentation code.

import cv2
import numpy as np

cap = cv2.VideoCapture(0)

while(1):
         


        
1条回答
  •  闹比i
    闹比i (楼主)
    2021-02-06 17:00

    Just add them together.

    import cv2
    import numpy as np
    
    cap = cv2.VideoCapture(0)
    
    while(1):
    
        # Take each frame
        _, frame = cap.read()
    
        # Convert BGR to HSV
        hsv = cv2.cvtColor(frame, cv2.COLOR_BGR2HSV)
    
        # define range of blue color in HSV
        lower_blue = np.array([110,50,50])
        upper_blue = np.array([130,255,255])
    
        lower_green = np.array([50, 50, 120])
        upper_green = np.array([70, 255, 255]) 
        green_mask = cv2.inRange(hsv, lower_green, upper_green) # I have the Green threshold image.
    
        # Threshold the HSV image to get only blue colors
        blue_mask = cv2.inRange(hsv, lower_blue, upper_blue)
        mask = blue_mask + green_mask
    
        # Bitwise-AND mask and original image
        res = cv2.bitwise_and(frame,frame, mask= mask)
    
        cv2.imshow('frame',frame)
        cv2.imshow('mask',mask)
        cv2.imshow('res',res)
        k = cv2.waitKey(5) & 0xFF
        if k == 27:
            break
    
    cv2.destroyAllWindows()
    

    0 讨论(0)
提交回复
热议问题