main.py

import cv2
import numpy as np

cap = cv2.VideoCapture('./data/video.mp4')
subtractor = cv2.createBackgroundSubtractorMOG2(history=100, varThreshold=50, detectShadows=True)

while(True):
    ret, frame = cap.read()
    
    mask = subtractor.apply(frame)
        
    kernel = np.ones((5,5),np.uint8)

    mask = cv2.morphologyEx(mask, cv2.MORPH_OPEN, kernel, iterations = 1)

    ret, mask = cv2.threshold(mask, 100, 255, cv2.THRESH_BINARY)

    mask = cv2.dilate(mask, kernel, iterations = 1)

    mask = cv2.morphologyEx(mask, cv2.MORPH_CLOSE, kernel,  iterations = 1)

    _, contours, h = cv2.findContours(mask, cv2.RETR_TREE, cv2.CHAIN_APPROX_SIMPLE)


    for cnt in contours:
        area =cv2.contourArea(cnt)
        #print(area)

        if area > 500:
            x,y,w,h = cv2.boundingRect(cnt)
            cv2.rectangle(frame,(x,y), (x+w, y+h), (0,255,0), 2)

    cv2.imshow('frame', frame)

    

    key = cv2.waitKey(30)
    if key ==ord('q'):
        break

cap.release()

cv2.destroyAllWindows()