OpenCV基于背景減除實現(xiàn)行人計數(shù)
前言
本文將使用OpenCV C++ 對視頻中的人流量進行統(tǒng)計。
一、圖像預處理

原圖如圖所示。本案例的需求是想要統(tǒng)計畫面中的人流量。畫面中走動的行人可以看作是前景,那么我們就需要將前景、背景分割出來。我們可以使用OpenCV提供的BackgroundSubtractorMOG2 高斯混合模型,將行人從畫面中分割出來,然后提取輪廓就可以統(tǒng)計人流量了。
Ptr<BackgroundSubtractorMOG2>MOG = createBackgroundSubtractorMOG2(); MOG->apply(frame, mask);

使用上面兩行代碼就可以創(chuàng)建高斯混合背景提取器。傳入原圖,返回背景減除結果。如上圖所示。接下來只需對上圖進行一些簡單操作,再提取輪廓就可以進行人流統(tǒng)計了。
threshold(mask, mask, 200, 255, THRESH_BINARY ); morphologyEx(mask, mask, MORPH_OPEN, kernel); dilate(mask, mask, kernel1);
進行二值化、形態(tài)學等操作可以將行人作為一個獨立個體分割出來。效果如圖。

二、對象計數(shù)
1.輪廓提取
將上面的二值圖像進行輪廓檢測,然后統(tǒng)計有效輪廓就可以完成對象計數(shù)了。
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++ 基于背景減除進行人流計數(shù),關鍵步驟有以下幾點。
1、使用BackgroundSubtractorMOG2 將前景從背景中分割出來。
2、將分割出來的前景進行輪廓提取,從而統(tǒng)計出人流量。
到此這篇關于OpenCV基于背景減除實現(xiàn)行人計數(shù)的文章就介紹到這了,更多相關OpenCV行人計數(shù)內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關文章希望大家以后多多支持腳本之家!
相關文章
C++獲取文件哈希值(hash)和獲取torrent(bt種子)磁力鏈接哈希值
這二個代碼一個是獲取文件哈希值的,另外一個是獲取torrent文件磁力鏈接的哈希值2013-11-11
VS2010 boost標準庫開發(fā)環(huán)境安裝教程
這篇文章主要為大家詳細介紹了VS2010 boost標準庫開發(fā)環(huán)境的安裝教程,具有一定的參考價值,感興趣的小伙伴們可以參考一下2017-04-04
C++實現(xiàn)產(chǎn)生隨機數(shù)和相應的猜拳小游戲?qū)嵗a
C++中沒有自帶的random函數(shù),要實現(xiàn)隨機數(shù)的生成就需要使用rand()和srand()。下面這篇文章主要給大家介紹了關于C++實現(xiàn)產(chǎn)生隨機數(shù)和相應的猜拳小游戲的相關資料,文中通過示例代碼介紹的非常詳細,需要的朋友可以參考下2018-09-09

