基于颜色的对象检测与跟踪

import cv2
import numpy as np


video = cv2.VideoCapture("E:\\video.mp4")
videoIsOpen=video.isOpened
print(videoIsOpen)
width=int(video.get(cv2.CAP_PROP_FRAME_WIDTH))#宽度
height=int(video.get(cv2.CAP_PROP_FRAME_HEIGHT))#高度
fps=video.get(cv2.CAP_PROP_FPS)#获取帧率
print(fps,width,height)

cv2.namedWindow('input video')
cv2.namedWindow('task video')

#外接矩形
def processRangle(sst,rect):
    findc,contours,hierarchy = cv2.findContours(sst,cv2.RETR_EXTERNAL,cv2.CHAIN_APPROX_SIMPLE) #获得图像的轮廓
    Size = len(contours)
    if Size>0 :
        maxArea=0.0
        for t in range(0,Size):
            area = cv2.contourArea(contours[t])#计算面积
            if area>maxArea:
                maxArea=area
                rect=cv2.boundingRect(contours[t])#矩形边框 返回值x,y,w,h[0-3]
    else:
        rect=0

kernel = cv2.getStructuringElement(cv2.MORPH_RECT,(3,3))
kerne2 = cv2.getStructuringElement(cv2.MORPH_RECT,(5,5))



while videoIsOpen:
    (flag,frame)=video.read()
    if not flag:
        break
    
    mask=cv2.inRange(frame,(0,127,0),(120,255,120))
    OPEND=cv2.morphologyEx(mask,cv2.MORPH_OPEN,kernel)#形态学操作
    dila = cv2.dilate(OPEND,kerne2)#膨胀
    cv2.imshow('task video',dila)

    roi = cv2.boundingRect(dila)

    processRangle(dila,roi)
    cv2.rectangle(frame,(roi[0],roi[1]),(roi[0]+roi[2],roi[1]+roi[3]),(0,0,255),5)
    cv2.imshow('input video',frame)

    c=cv2.waitKey(40)
    if c==27:
        break


video.release()
cv2.waitKey(0)