视频绿幕抠图程序源码MFCApplication1.rar

 // MFCApplication1Dlg.cpp: 实现文件 // #include "pch.h" #include "framework.h" #include "MFCApplication1.h" #include "MFCApplication1Dlg.h" #include "afxdialogex.h" #include <iostream> #include "opencv2/highgui/highgui.hpp" #include "opencv2/imgproc/imgproc.hpp" #include "opencv2/imgproc/imgproc_c.h" #include "opencv2/core/core.hpp" #include "opencv2/objdetect/objdetect.hpp" #include <opencv2/face.hpp> #include <opencv2/xfeatures2d.hpp> #include "opencv2/opencv.hpp" #include <math.h> #include<direct.h> #include<io.h> #include<sstream> #include "string" #ifdef _DEBUG #define new DEBUG_NEW #endif using namespace std; using namespace cv; using namespace cv::face; using namespace cv::xfeatures2d; CFont font, font1; char c; // 用于应用程序“关于”菜单项的 CAboutDlg 对话框 class CAboutDlg : public CDialogEx { public: CAboutDlg(); // 对话框数据 #ifdef AFX_DESIGN_TIME enum { IDD = IDD_ABOUTBOX }; #endif protected: virtual void DoDataExchange(CDataExchange* pDX); // DDX/DDV 支持 // 实现 protected: DECLARE_MESSAGE_MAP() }; CAboutDlg::CAboutDlg() : CDialogEx(IDD_ABOUTBOX) { } void CAboutDlg::DoDataExchange(CDataExchange* pDX) { CDialogEx::DoDataExchange(pDX); } BEGIN_MESSAGE_MAP(CAboutDlg, CDialogEx) END_MESSAGE_MAP() // CMFCApplication1Dlg 对话框 CMFCApplication1Dlg::CMFCApplication1Dlg(CWnd* pParent /*=nullptr*/) : CDialogEx(IDD_MFCAPPLICATION1_DIALOG, pParent) { m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME); } void CMFCApplication1Dlg::DoDataExchange(CDataExchange* pDX) { CDialogEx::DoDataExchange(pDX); } BEGIN_MESSAGE_MAP(CMFCApplication1Dlg, CDialogEx) ON_WM_SYSCOMMAND() ON_WM_PAINT() ON_WM_QUERYDRAGICON() ON_BN_CLICKED(IDC_BUTTON1, &CMFCApplication1Dlg::OnBnClickedButton1) ON_BN_CLICKED(IDC_BUTTON6, &CMFCApplication1Dlg::OnBnClickedButton6) END_MESSAGE_MAP() // CMFCApplication1Dlg 消息处理程序 BOOL CMFCApplication1Dlg::OnInitDialog() { CDialogEx::OnInitDialog(); // 将“关于...”菜单项添加到系统菜单中。 // IDM_ABOUTBOX 必须在系统命令范围内。 ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX); ASSERT(IDM_ABOUTBOX < 0xF000); CMenu* pSysMenu = GetSystemMenu(FALSE); if (pSysMenu != nullptr) { BOOL bNameValid; CString strAboutMenu; bNameValid = strAboutMenu.LoadString(IDS_ABOUTBOX); ASSERT(bNameValid); if (!strAboutMenu.IsEmpty()) { pSysMenu->AppendMenu(MF_SEPARATOR); pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu); } } // 设置此对话框的图标。 当应用程序主窗口不是对话框时，框架将自动 // 执行此操作 SetIcon(m_hIcon, TRUE); // 设置大图标 SetIcon(m_hIcon, FALSE); // 设置小图标 ShowWindow(SW_MAXIMIZE); ShowWindow(SW_MINIMIZE); // TODO: 在此添加额外的初始化代码 //CFont font; font.CreatePointFont(200, _T("宋体"), NULL); font1.CreatePointFont(150, _T("宋体"), NULL); GetDlgItem(IDC_BUTTON1)->SetFont(&font); GetDlgItem(IDC_BUTTON6)->SetFont(&font); GetDlgItem(IDC_STATIC12)->SetFont(&font1); GetDlgItem(IDC_STATIC13)->SetFont(&font1); return TRUE; // 除非将焦点设置到控件，否则返回 TRUE } void CMFCApplication1Dlg::OnSysCommand(UINT nID, LPARAM lParam) { if ((nID & 0xFFF0) == IDM_ABOUTBOX) { CAboutDlg dlgAbout; dlgAbout.DoModal(); } else { CDialogEx::OnSysCommand(nID, lParam); } } // 如果向对话框添加最小化按钮，则需要下面的代码 // 来绘制该图标。 对于使用文档/视图模型的 MFC 应用程序， // 这将由框架自动完成。 void CMFCApplication1Dlg::OnPaint() { if (IsIconic()) { CPaintDC dc(this); // 用于绘制的设备上下文 SendMessage(WM_ICONERASEBKGND, reinterpret_cast<WPARAM>(dc.GetSafeHdc()), 0); // 使图标在工作区矩形中居中 int cxIcon = GetSystemMetrics(SM_CXICON); int cyIcon = GetSystemMetrics(SM_CYICON); CRect rect; GetClientRect(&rect); int x = (rect.Width() - cxIcon + 1) / 2; int y = (rect.Height() - cyIcon + 1) / 2; // 绘制图标 dc.DrawIcon(x, y, m_hIcon); } else { CDialogEx::OnPaint(); } } //当用户拖动最小化窗口时系统调用此函数取得光标 //显示。 HCURSOR CMFCApplication1Dlg::OnQueryDragIcon() { return static_cast<HCURSOR>(m_hIcon); } CWinThread* pWinThread_CompositeVideo;//创建线程 string convertToString(double d) { ostringstream os; if (os << d) return os.str(); return "invalid conversion"; } UINT CMFCApplication1Dlg::CompositeVideo(LPVOID pParam)//**************************************************************合成视频线程函数****************************************************************************************** { CMFCApplication1Dlg * dlgR = (CMFCApplication1Dlg *)pParam; // TODO: 在此添加控件通知处理程序代码 //绿幕抠图 VideoWriter writer; //writer.open("./my1.AVI", writer.fourcc('M', 'J', 'P', 'G'), 10, Size(1366,768), true);//CAP_OPENCV_MJPEG writer.open("./my1.flv", writer.fourcc('F', 'L', 'V', '1'), 30, Size(1366, 768), true);//CAP_OPENCV_MJPEG Mat background_01 = imread("D:/OpenCV_program/绿幕抠图/1.jpg"); Mat background_02 = imread("D:/OpenCV_program/绿幕抠图/2.jpg"); Mat background_03 = imread("D:/OpenCV_program/绿幕抠图/3.jpg"); Mat background_04 = imread("D:/OpenCV_program/绿幕抠图/4.jpg"); Mat background = background_01; int width = background.cols; int height = background.rows; VideoCapture capture, capture_TV; capture.open("./胆小的猫.flv"); capture_TV.open("./跳舞.mp4"); if ((!capture.isOpened()) || (!capture_TV.isOpened())) { return -1; } Mat frame, hsv, green_mask, green_mask_bin, mask, result; Mat k, k0; int m = 0; int h, w, dims; int r = 0, g = 0, b = 0; int r1 = 0, g1 = 0, b1 = 0; int r2 = 0, g2 = 0, b2 = 0; double wt = 0; vector<Vec4i> hierarchy; vector<vector<Point> > contours; while (capture.read(frame)) { mask = Mat::zeros(frame.size(), CV_8UC1); result = Mat::zeros(frame.size(), frame.type()); //寻找绿幕 cvtColor(frame, hsv, COLOR_BGR2HSV); inRange(hsv, Scalar(35, 43, 46), Scalar(77, 255, 255), green_mask);//大范围的绿色 //inRange(hsv, Scalar(50, 200, 200), Scalar(70, 255, 255), green_mask);//本素材的绿色 //imshow("green_mask", green_mask); // 开操作 //k0 = getStructuringElement(MORPH_RECT, Size(5, 5), Point(-1, -1)); //morphologyEx(green_mask, green_mask, MORPH_OPEN, k0); // 闭操作，去掉mask杂点 //k = getStructuringElement(MORPH_RECT, Size(3, 3), Point(-1, -1)); //morphologyEx(green_mask, green_mask, MORPH_CLOSE, k); //mask二值化 threshold(green_mask, green_mask_bin, 200, 255, THRESH_BINARY); //寻找轮廓 findContours(green_mask_bin, contours, hierarchy, RETR_TREE, CHAIN_APPROX_NONE, Point(0, 0)); //筛选剔除掉面积小于290000或大于320000的轮廓 vector <vector<Point>>::iterator iter = contours.begin(); for (; iter != contours.end();) { double g_dConArea = contourArea(*iter); if (g_dConArea < 290000 || g_dConArea > 320000) { iter = contours.erase(iter); } else { ++iter; } } if (contours.size() == 0) { frame.copyTo(result); imshow("ImageShow2", frame); imshow("ImageShow", result); writer.write(result); } else { capture_TV.read(background); drawContours(mask, contours, -1, Scalar(255), CV_FILLED); for (int i = 0; i < (int)contours.size(); i++) { Rect rect1 = boundingRect(Mat(contours[i]));//最小外接矩 resize(background, background, rect1.size());//调整图像大小 //融合 h = frame.rows; w = frame.cols; //int bg_row = 0; //dims = frame.channels();