前言
这篇博客针对《Python+OpenCV混合高斯建模算法人体识别出入口人流量统计计数》编写代码，代码整洁，规则，易读。 学习与应用推荐首选。
运行结果
文章目录
一、所需工具软件
二、使用步骤
1. 主要代码
2. 运行结果
三、在线协助
一、所需工具软件
1. Python
2. Pycharm
二、使用步骤
代码如下（示例）：
import numpy as np
import math
import cv2
cap = cv2.VideoCapture('video02.mp4')
# fgbg = cv2.createBackgroundSubtractorMOG2(history=5, varThreshold=150)
def line1(x,y):
return y - (29*x)/96.0 - 300
crossedAbove = 0
crossedBelow = 0
points = set()
pointFromAbove = set()
pointFromBelow = set()
fourcc = cv2.VideoWriter_fourcc(*'XVID')
out = cv2.VideoWriter('pedestrianOutput.avi',fourcc, 25.0, (1920,1080))
font = cv2.FONT_HERSHEY_SIMPLEX
while(1):
pointInMiddle = set()
prev = points
points = set()
ret, frame = cap.read()
fgmask = cv2.blur(frame, (4,4))
fgmask = fgbg.apply(fgmask)
oldFgmask = fgmask.copy()
oldFgmask,contours, hierarchy = cv2.findContours(fgmask, cv2.RETR_EXTERNAL,1)
for contour in contours:
x,y,w,h = cv2.boundingRect(contour)
cv2.rectangle(frame,(x,y),(x+w,y+h),(0,255,0),6, lineType=cv2.LINE_AA)
point = (int(x+w/2.0), int(y+h/2.0))
points.add(point)
for point in points:
(xnew, ynew) = point
if line1(xnew, ynew) > 0 and line2(xnew, ynew) < 0:
pointInMiddle.add(point)
for prevPoint in prev:
(xold, yold) = prevPoint
dist = cv2.sqrt((xnew-xold)*(xnew-xold)+(ynew-yold)*(ynew-yold))
pointFromAbove.add(point)
elif line2(xold, yold) > 0: # Point entered from line below
pointFromBelow.add(point)
else: # Point was inside the block
if prevPoint in pointFromBelow:
pointFromBelow.remove(prevPoint)
pointFromBelow.add(point)
elif prevPoint in pointFromAbove:
pointFromAbove.remove(prevPoint)
pointFromAbove.add(point)
if line1(xnew, ynew) < 0 and prevPoint in pointFromBelow: # Point is above the line
print('One Crossed Above')
print(point)
crossedAbove += 1
pointFromBelow.remove(prevPoint)
if line2(xnew, ynew) > 0 and prevPoint in pointFromAbove: # Point is below the line
print('One Crossed Below')
print(point)
crossedBelow += 1
for point in points:
if point in pointFromBelow:
cv2.circle(frame, point, 3, (255,0,255),6)
elif point in pointFromAbove:
cv2.circle(frame, point, 3, (0,0,255),6)
cv2.line(frame, (0,300), (1920,880), (255, 0, 0), 4)
cv2.line(frame, (0,500), (1920,1080), (255, 0, 0), 4)
cv2.putText(frame,'People Going Above = '+str(crossedAbove),(800,50), font, 2,(255,255,255),5,cv2.LINE_AA)
cv2.putText(frame,'People Going Below = '+str(crossedBelow),(800,100), font, 2,(255,255,255),5,cv2.LINE_AA)
cv2.putText(frame,'People Going Total = '+str(crossedBelow+crossedAbove),(800,150), font, 2,(255,255,255),5,cv2.LINE_AA)
cv2.namedWindow("frame", 0);
cv2.resizeWindow("frame", 540, 380);
cv2.moveWindow("frame",100,100)
#cv2.imshow('oldFgmask',oldFgmask)
cv2.imshow('frame',frame)
out.write(frame)
l = cv2.waitKey(1) & 0xff
if l == 27:
break
cap.release()
cv2.destroyAllWindows()
运行结果
三、在线协助：
如需安装运行环境或远程调试，见文章底部个人 QQ 名片，由专业技术人员远程协助！
当前文章连接：https://blog.csdn.net/alicema1111/article/details/132666851
个人博客主页：https://blog.csdn.net/alicema1111?type=blog
博主所有文章点这里：https://blog.csdn.net/alicema1111?type=blog
博主推荐：
Python人脸识别考勤打卡系统：
https://blog.csdn.net/alicema1111/article/details/133434445
Python果树水果识别：https://blog.csdn.net/alicema1111/article/details/130862842
Python+Yolov8+Deepsort入口人流量统计：https://blog.csdn.net/alicema1111/article/details/130454430
Python+Qt人脸识别门禁管理系统：https://blog.csdn.net/alicema1111/article/details/130353433
Python+Qt指纹录入识别考勤系统：https://blog.csdn.net/alicema1111/article/details/129338432
Python Yolov5火焰烟雾识别源码分享：https://blog.csdn.net/alicema1111/article/details/128420453
Python+Yolov8路面桥梁墙体裂缝识别：https://blog.csdn.net/alicema1111/article/details/133434445