python+opencv 一个简单地检查课堂人头数的程序

Python+Opencv 一个简单地检查课堂人头数的程序

先保存一张教室图片：

然后用鼠标左键选择ROI区域

if event == cv.EVENT_LBUTTONDOWN:  # ---- 左键点击：选择点

按一下滚轮键可以选择下一个ROI

  if event == cv.EVENT_MBUTTONDOWN:  # ---- 中键点击：选择点

左键双击完成选择

if event == cv.EVENT_LBUTTONDBLCLK:  # --- 双击: 结束选取

通过把ROI以外的区域变成白色 以内的区域进行 模糊 - 二值化 - 开闭运算 - 计算连通区域
然后大致会计算教室的人数

人数就是白色区域的个数

**本人还是在学习阶段 这种方法误差大 还请各位大佬多多指教 **

import cv2 as cv
import numpy as np


# ------------------ 鼠标响应函数 ----------------------
def mouse_chose_mask(event, x, y, flags, param):
    global img   # 原图片
    global lsPointsChoose, tpPointsChoose  # 存入选择的点 ls用于提取 tp用于画线
    global pointsCount  # 对鼠标按下的点计数
    global img2   # 传入图片的副本
    global flag   # 交换点标志位

    img2 = img.copy()  # 保证每次都重新再原图画  避免画多了

    if event == cv.EVENT_LBUTTONDOWN:  # ---- 左键点击：选择点
        pointsCount = pointsCount + 1
        point1 = (x, y)  # 用于画圆的点
        cv.circle(img2, point1, 10, (0, 255, 0), 2)  # 画圆便于提醒

        # 将选取的点保存到list列表里
        lsPointsChoose.append([x, y])  # 用于转化为数组 提取多边形ROI
        tpPointsChoose.append((x, y))  # 用于画点

        if len(flag) != 1:  # 如果是多个ROI
            # print(flag)
            for k in range(len(flag) - 1):
                # print(flag[k])
                # print(flag[k+1])
                cv.line(img2, tpPointsChoose[flag[k]], tpPointsChoose[flag[k+1]-1], (0, 0, 255), 2)

            # 多个RO