Connected Components in Python

connected_components.py

import cv2
import numpy as np

def connected_components(thresh_img):
    thresh_img = mog_mask.copy()
    contours, hierarchy = cv2.findContours(thresh_img, cv2.RETR_TREE,
                                           cv2.CHAIN_APPROX_SIMPLE)
    fg_mask = np.zeros(frame.shape[:2], np.uint8)
    i = 0
    cs = []
    for c in contours:
        cs.append(len(c))
    cmax = max(cs)
    cavg = np.mean(cs)
    cstd = np.std(cs)
    i = 0
    for c in contours:
        if len(c) > cmax - 3 * cstd:
            cv2.drawContours(fg_mask, [c], 0, (1, 1, 1), -1)
        i += 1
    return fg_mask