玉米识别背景建模

import numpy as np
import cv2

cap = cv2.VideoCapture('../yu1.MP4')
#cap = cv2.VideoCapture('../yumi.MP4')
#结构元素提取
kernel = cv2.getStructuringElement(cv2.MORPH_ELLIPSE,(3,3))
#创建混合高斯模型用于背景建模
fgbg = cv2.createBackgroundSubtractorMOG2()

yumi = []
zhen = 0
i = 0
yumi_num = 0
while(True):
    zhen+=1
    ret, frame = cap.read()
    fgmask = fgbg.apply(frame)
    #形态学开运算去噪点
    fgmask = cv2.morphologyEx(fgmask, cv2.MORPH_OPEN, kernel)
    ret, fgmask = cv2.threshold(fgmask, 10, 255, cv2.THRESH_BINARY)
    #寻找视频中的轮廓
    im, contours, hierarchy = cv2.findContours(fgmask, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
    cv2.line(frame, (10, 550), (1200, 550), (0, 255, 255), 3)
    # cv2.line(frame, (10, 850), (1200, 850), (0, 255, 255), 3)
    contours_num = 0
    for c in contours:
        #计算各轮廓的周长
        perimeter = cv2.arcLength(c,True)
        x, y, w, h = 0, 0,0 ,0
        if 1000 > perimeter > 50:
            #找到一个直矩形（不会旋转）
            x,y,w,h = cv2.boundingRect(c)
            contours_num +=1
            if contours_num==1:
                i+=1
                cv2.imwrite(str(i)+'.png' ,frame,params=None)
            # 画出这个矩形
            cv2.rectangle(frame,(x,y),(x+w,y+h),(0,255,0),2)


        #保存中心点信息
        cv2.rectangle(frame, (x, y), (x + w, y + h), (0, 0, 255), 2)
        centre_p = (x + int(w / 2), y + int(h / 2))
        yumi.append(centre_p)
        cv2.circle(frame, (centre_p), 5, (0, 0, 255), -1)
        for (x, y) in yumi:
            if (550 < (y + int(h / 2)) < 850):
                yumi_num += 1
                yumi.remove((x, y))
    cv2.putText(frame, "yumi Count:" + str(yumi_num), (100, 60), cv2.FONT_HERSHEY_SIMPLEX, 2, (0, 0, 255), 5)

    cv2.imshow('frame',frame)
    cv2.imshow('fgmask', fgmask)
    k = cv2.waitKey(50) & 0xff
    if k == 27:
        break
print(frame.shape)
cap.release()
cv2.destroyAllWindows()