#include<opencv2/opencv.hpp>
#include<iostream>
using namespace cv;
using namespace std;
int main(int argc, char **argv) {
VideoCapture capture;
capture.open("D:/picture/opencv/images/video_003.avi");
if (!capture.isOpened()) {
printf("could not load video file...\n");
return -1;
}
Mat frame;
Mat bsmaskMOG2;
Mat bsmaskKNN;
namedWindow("Video-demo", CV_WINDOW_AUTOSIZE);
namedWindow("MOG2", CV_WINDOW_AUTOSIZE);
namedWindow("KNN Model", CV_WINDOW_AUTOSIZE);
Mat kernel = getStructuringElement(MORPH_RECT, Size(3, 3), Point(-1, -1));
//intialization BS
Ptr<BackgroundSubtractor> pMOG2 = createBackgroundSubtractorMOG2();
Ptr<BackgroundSubtractor> pKNN = createBackgroundSubtractorKNN();
while (capture.read(frame)) {
imshow("Video-demo", frame);
//MOG2 BSM
pMOG2->apply(frame, bsmaskMOG2);
Mat kernel = getStructuringElement(MORPH_RECT, Size(3, 3), Point(-1, -1));
morphologyEx(bsmaskMOG2, bsmaskMOG2, MORPH_OPEN, kernel, Point(-1, -1));
imshow("MOG2", bsmaskMOG2);
//KNN BSM
pKNN->apply(frame, bsmaskKNN,-1);
imshow("KNN Model", bsmaskKNN);
//show mask
char c = waitKey(100);
if (c == 27)
break;
}
capture.release();
waitKey(0);
return 0;
}
相关文章: