playvideo.cpp

#include <opencv2/core/core.hpp>
#include <opencv2/highgui/highgui.hpp>
#include <opencv2/imgproc/imgproc.hpp>
#include <opencv2/video/background_segm.hpp>
#include <opencv2/video/tracking.hpp>
#include <opencv/cv.hpp>
#include <iostream>
#include <sstream>
#include <string.h>
#include <stdio.h>

using namespace cv;
using namespace std;

int main(void) {
    cv::VideoCapture capVideo ("mobilku.mp4");
    cv::Mat imgFrame;
    cv::Mat grayscaleFrame;
    cv::Ptr<cv::BackgroundSubtractor> bg_model = cv::createBackgroundSubtractorMOG2().dynamicCast<cv::BackgroundSubtractor>();

    if (!capVideo.isOpened()) {                                                 // if unable to open video file
        std::cout << "\nerror reading video file" << std::endl << std::endl;      // show error message
        return(0);                                                              // and exit program
    }
	cv::namedWindow( "video", 1);
    for( ; ; )
    {
		cv::Mat frame, resized;
        capVideo >> frame;
        if(frame.empty()) break;
        //imshow("video", frame);
        //waitKey(10);
		cv::resize(frame, resized, cv::Size(800, 600));
		//cvLine(resized,center,center1,Scalar(255, 0, 0),1 CV_AA);
		line(resized, Point(150,240), Point(750,240), Scalar(0, 0, 255), 2, 8, 0);
		cvtColor(frame, grayscaleFrame, CV_BGR2GRAY);
    	equalizeHist(grayscaleFrame, grayscaleFrame);
    	cv::Mat grayFrame, foregroundMask, foregroundFrame, backgroundFrame;
//	cv::namedWindow("grayFrame",1);
//  cv::imshow("grayFrame", grayscaleFrame);
//  cv::waitKey(20); // waits to display frame
		if( foregroundMask.empty() ){
    	    foregroundMask.create(resized.size(), resized.type());
    	    }
		bg_model->apply(resized, foregroundMask, true ? -1 : 0);
		cv::GaussianBlur(foregroundMask, foregroundMask, cv::Size(11,11), 3.5,3.5);
		cv::threshold(foregroundMask, foregroundMask, 130, 255, cv::THRESH_BINARY);
		foregroundFrame = cv::Scalar::all(0);
		resized.copyTo(foregroundFrame, foregroundMask);
		bg_model->getBackgroundImage(backgroundFrame);
    	Mat Temp, img;
		foregroundMask.convertTo(Temp, CV_8UC1);
		bilateralFilter(Temp, img, 5, 20, 20);
		vector<vector<Point> > contours;
		vector<Vec4i> hier;
    	cv::findContours(Temp, contours,CV_RETR_EXTERNAL,CV_CHAIN_APPROX_SIMPLE);
    	cv::Mat contourImage(Temp.size(), CV_8UC3, cv::Scalar(0,0,0));
		for (size_t idx = 0; idx < contours.size(); idx++) {
			cv::drawContours(contourImage, contours, idx, Scalar(255, 255, 255), 2, 8, hier);
			Rect box = boundingRect(contours[idx]); 
			rectangle(resized, box, Scalar(0,255,0));
			//circle(contourImage, Point(50,50),50, Scalar(255,255,255),CV_FILLED, 8,0);
		}

		imshow("resized",resized);
		imshow("foreground mask", foregroundMask);
    	imshow("foreground frame", foregroundFrame);
    	imshow("Contour", contourImage);
		cv::waitKey(10);
    }
    cv::waitKey(0); 
    return(0);
}