我如何计算每帧中的边界框？

Question

我需要知道如何计算每帧中的边界框？

我需要知道总边界框，所以我知道何时需要添加要跟踪的新对象，何时total bounding boxes in current frame > total bounding boxes in previous frame

我已经尝试将商店质心坐标（cx，cy）放入列表中：

import cv2
import numpy as np

bgsMOG    = cv2.BackgroundSubtractorMOG(50,3,0.8)
cap       = cv2.VideoCapture("d:\\MOV_5702.avi")
a         = []


if cap:
    while True:
        ret, frame = cap.read()
        if ret:
            fgmask              = bgsMOG.apply(frame, None, 0.99)
            contours, hierarchy = cv2.findContours(fgmask, cv2.RETR_LIST,cv2.CHAIN_APPROX_SIMPLE)
            cv2.drawContours(frame,contours,-1,(0,255,0),cv2.cv.CV_FILLED,32)
            try: hierarchy = hierarchy[0]
            except: hierarchy = []
            for contour, hier in zip(contours, hierarchy):
                (x,y,w,h) = cv2.boundingRect(contour)

                if w > 20 and h > 20:
                    cv2.rectangle(frame, (x,y), (x+w,y+h), (180,0,0), 1)
                    (x,y,w,h) = cv2.boundingRect(contour)

                    x1=w/2
                    y1=h/2
                    cx=x+x1
                    cy=y+y1
                    a.append([cx,cy])
                    print(len(a))
                    a=[]

            cv2.imshow('BGS', fgmask)
            cv2.imshow('Ori+Bounding Box',frame)
            key = cv2.waitKey(100)
            if key == ord('q'):
                break
cap.release()
cv2.destroyAllWindows()

但结果总是1 ..

Answer 1

在您的代码中，您有

a.append([cx,cy])
print(len(a))
a=[]

由于这是在循环中运行，因此在添加一个项目后，a始终会重置为空列表（因此len(a)始终为1）。

尝试拉a=[]循环外的for，使其每帧只发生一次，而不是每个轮廓发生一次，它应该更好地指示每个边界框的数量帧。