OpenCV基于背景减除实现行人计数

更新时间：2022年01月12日 14:43:47 作者：Zero___Chen

本文主要介绍了如何使用OpenCV C++对视频中的人流量进行统计。文中的示例代码讲解详细，对我们学习OpenCV有一定的帮助，需要的可以了解一下

前言

本文将使用OpenCV C++ 对视频中的人流量进行统计。

一、图像预处理

原图如图所示。本案例的需求是想要统计画面中的人流量。画面中走动的行人可以看作是前景，那么我们就需要将前景、背景分割出来。我们可以使用OpenCV提供的BackgroundSubtractorMOG2 高斯混合模型，将行人从画面中分割出来，然后提取轮廓就可以统计人流量了。

Ptr<BackgroundSubtractorMOG2>MOG = createBackgroundSubtractorMOG2();
MOG->apply(frame, mask);

使用上面两行代码就可以创建高斯混合背景提取器。传入原图，返回背景减除结果。如上图所示。接下来只需对上图进行一些简单操作，再提取轮廓就可以进行人流统计了。

threshold(mask, mask, 200, 255, THRESH_BINARY );

morphologyEx(mask, mask, MORPH_OPEN, kernel);

dilate(mask, mask, kernel1);

进行二值化、形态学等操作可以将行人作为一个独立个体分割出来。效果如图。

二、对象计数

1.轮廓提取

将上面的二值图像进行轮廓检测，然后统计有效轮廓就可以完成对象计数了。

    vector<vector<Point>>contours;
    vector<vector<Point>>EffectiveContours;    
    findContours(mask, contours, RETR_EXTERNAL, CHAIN_APPROX_SIMPLE);
    for (int i = 0; i < contours.size(); i++)
    {            
        double area = contourArea(contours[i]);
    
        if (area > 300)
        {        
            EffectiveContours.push_back(contours[i]);
        }
        
    }

2.效果显示

	char text[10];
	for (int i = 0; i < EffectiveContours.size(); i++)
	{
		RotatedRect rect = minAreaRect(EffectiveContours[i]);

		Rect box = rect.boundingRect();

		rectangle(frame, Rect(box.x, box.y, box.width, box.height), Scalar(0, 255, 0), 2);

		sprintf_s(text, "%s%d", "Current:", EffectiveContours.size());

		putText(frame, text, Point(10, 30), FONT_HERSHEY_SIMPLEX, 1, Scalar(0, 255, 0), 2);
	}

最终效果如图所示。

三、源码

#include<iostream>
#include<opencv2/opencv.hpp>
using namespace std;
using namespace cv;

int main()
{

	VideoCapture capture;
	capture.open("1.avi");

	if (!capture.isOpened())
	{
		cout << "Can not open video source!" << endl;
		system("pause");
		return -1;
	}

	Ptr<BackgroundSubtractorMOG2>MOG = createBackgroundSubtractorMOG2();

	Mat kernel = getStructuringElement(MORPH_RECT, Size(3, 5));
	Mat kernel1 = getStructuringElement(MORPH_RECT, Size(7, 3));

	Mat frame, mask;
	while (capture.read(frame))
	{
		MOG->apply(frame, mask);

		threshold(mask, mask, 200, 255, THRESH_BINARY );

		morphologyEx(mask, mask, MORPH_OPEN, kernel);

		dilate(mask, mask, kernel1);

		vector<vector<Point>>contours;
		vector<vector<Point>>EffectiveContours;	
		findContours(mask, contours, RETR_EXTERNAL, CHAIN_APPROX_SIMPLE);
		for (int i = 0; i < contours.size(); i++)
		{			
			double area = contourArea(contours[i]);

			if (area > 300)
			{		
				EffectiveContours.push_back(contours[i]);
			}
			
		}

		char text[10];
		for (int i = 0; i < EffectiveContours.size(); i++)
		{
			RotatedRect rect = minAreaRect(EffectiveContours[i]);

			Rect box = rect.boundingRect();

			rectangle(frame, Rect(box.x, box.y, box.width, box.height), Scalar(0, 255, 0), 2);

			sprintf_s(text, "%s%d", "Current:", EffectiveContours.size());

			putText(frame, text, Point(10, 30), FONT_HERSHEY_SIMPLEX, 1, Scalar(0, 255, 0), 2);
		}
		imshow("frame", frame);
		imshow("mask", mask);

		char key = waitKey(10);
		if (key == 27)
		{
			break;
		}
	}

	destroyAllWindows();
	capture.release();
	system("pause");
	return 0;
}

总结

本文使用OpenCV C++ 基于背景减除进行人流计数，关键步骤有以下几点。

1、使用BackgroundSubtractorMOG2 将前景从背景中分割出来。

2、将分割出来的前景进行轮廓提取，从而统计出人流量。

到此这篇关于OpenCV基于背景减除实现行人计数的文章就介绍到这了,更多相关OpenCV行人计数内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！

您可能感兴趣的文章:

C++11中多线程编程-std::async的深入讲解
这篇文章主要给大家介绍了关于C++11中多线程编程-std::async的相关资料，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
2020-11-11
C++ 如何将Lambda转换成函数指针
这篇文章主要介绍了C++ 如何将Lambda转换成函数指针，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教
2022-11-11
Qt数据库应用之实现通用数据库采集
这篇文章主要为大家介绍了Qt中是如何实现通用数据库采集的，文中的示例代码讲解详细，对我们学习Qt有一定帮助，感兴趣的小伙伴可以了解一下
2022-03-03
C++判断一个点是否在圆内的方法
这篇文章主要为大家详细介绍了C++判断一个点是否在圆内的方法，具有一定的参考价值，感兴趣的小伙伴们可以参考一下
2018-05-05
C语言实现俄罗斯方块的六种模式详程建议收藏
迟早一定会挂掉的俄罗斯方块，为什么至今仍是世界游戏之王？它是怎么编写的？本文将给大家详细介绍六种模式的实现，对大家的学习或工作具有一定的参考借鉴价值
2022-02-02
C/C++中的名字空间与作用域示例详解
这篇文章主要给大家介绍了关于C/C++中名字空间与作用域的相关资料，文中通过示例代码介绍的非常详细，对大家学习或者使用c/c++具有一定的参考学习价值，需要的朋友们下面随着小编来一起看看吧。
2017-09-09
基于ROS 服务通信模式详解
今天小编就为大家分享一篇基于ROS 服务通信模式详解，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
2019-12-12
八皇后问题实现代码分享
八皇后问题，是一个古老而著名的问题，是回溯算法的典型案例,这篇文章主要介绍了八皇后问题实现代码,需要的朋友可以参考下
2014-02-02
通过C++程序示例理解设计模式中的外观模式
这篇文章主要介绍了通过设计模式中的外观模式及相关的C++程序示例,外观模式在高层提供了一个统一的接口实现一定程度上的解耦,需要的朋友可以参考下
2016-03-03
利用C语言实现三子棋(井字棋)小游戏
这篇文章主要为大家详细介绍了利用C语言实现三子棋小游戏，文中示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下
2021-08-08