使用OpenCV實現(xiàn)檢測和追蹤車輛
更新時間:2018年01月09日 10:50:19 作者:炒雞嗨客協(xié)管徐
這篇文章主要為大家詳細(xì)介紹了使用OpenCV實現(xiàn)檢測和追蹤車輛,具有一定的參考價值,感興趣的小伙伴們可以參考一下
本文實例為大家分享了OpenCV實現(xiàn)檢測和追蹤車輛的具體代碼,供大家參考,具體內(nèi)容如下

完整源碼GitHub
- 使用高斯混合模型(BackgroundSubtractorMOG2)對背景建模,提取出前景
- 使用中值濾波去掉椒鹽噪聲,再閉運算和開運算填充空洞
- 使用cvBlob庫追蹤車輛,我稍微修改了cvBlob源碼來通過編譯
由于要對背景建模,這個方法要求背景是靜止的
另外不同車輛白色區(qū)域不能連通,否則會認(rèn)為是同一物體
void processVideo(char* videoFilename)
{
Mat frame; // current frame
Mat fgMaskMOG2; // fg mask fg mask generated by MOG2 method
Mat bgImg; // background
Ptr<BackgroundSubtractorMOG2> pMOG2 = createBackgroundSubtractorMOG2(200, 36.0, false); // MOG2 Background subtractor
while (true)
{
VideoCapture capture(videoFilename);
if (!capture.isOpened())
{
cerr << "Unable to open video file: " << videoFilename << endl;
return;
}
int width = (int)capture.get(CV_CAP_PROP_FRAME_WIDTH);
int height = (int)capture.get(CV_CAP_PROP_FRAME_HEIGHT);
unique_ptr<IplImage, void(*)(IplImage*)> labelImg(cvCreateImage(cvSize(width, height), IPL_DEPTH_LABEL, 1),
[](IplImage* p){ cvReleaseImage(&p); });
CvBlobs blobs;
CvTracks tracks;
while (true)
{
// read input data. ESC or 'q' for quitting
int key = waitKey(1);
if (key == 'q' || key == 27)
return;
if (!capture.read(frame))
break;
// update background
pMOG2->apply(frame, fgMaskMOG2);
pMOG2->getBackgroundImage(bgImg);
imshow("BG", bgImg);
imshow("Original mask", fgMaskMOG2);
// post process
medianBlur(fgMaskMOG2, fgMaskMOG2, 5);
imshow("medianBlur", fgMaskMOG2);
morphologyEx(fgMaskMOG2, fgMaskMOG2, MORPH_CLOSE, getStructuringElement(MORPH_RECT, Size(5, 5))); // fill black holes
morphologyEx(fgMaskMOG2, fgMaskMOG2, MORPH_OPEN, getStructuringElement(MORPH_RECT, Size(5, 5))); // fill white holes
imshow("morphologyEx", fgMaskMOG2);
// track
cvLabel(&IplImage(fgMaskMOG2), labelImg.get(), blobs);
cvFilterByArea(blobs, 64, 10000);
cvUpdateTracks(blobs, tracks, 10, 90, 30);
cvRenderTracks(tracks, &IplImage(frame), &IplImage(frame));
// show
imshow("Frame", frame);
key = waitKey(30);
}
}
}
以上就是本文的全部內(nèi)容,希望對大家的學(xué)習(xí)有所幫助,也希望大家多多支持腳本之家。
相關(guān)文章
高效實現(xiàn)整型數(shù)字轉(zhuǎn)字符串int2str的方法
下面小編就為大家?guī)硪黄咝崿F(xiàn)整型數(shù)字轉(zhuǎn)字符串int2str的方法。小編覺得挺不錯的,現(xiàn)在就分享給大家,也給大家做個參考。一起跟隨小編過來看看吧2017-03-03

