OpenCV图像分割实战C++（一）Grabcut抠图与证件照背景替换_opencv 视频图片抠图换背景

Grabcut抠图

步骤：

• 输入原图像
• 矩形输入
• 开始分类
• GMM描述
• GMM训练分类
• Graph Cut分类
• 不断迭代直到收敛分类

API：

void grabCut(
InputArray img, // 待分割图像，8bit，3通道
// 输入输出参数，保存处理后的结果，8bit单通道掩码(与img同rows cols)，mask元素值只能为 GC_BGD, GC_FGD, GC_PR_BGD, GC_PR_FGD 之一
InputOutputArray mask, // 如果没有手动标记 GC_BGD或GC_FGD ，那么结果只会有 GC_PR_BGD或GC_PR_FGD
Rect rect, // 当 mode=GC_INIT_WITH_RECT时使用，rect外部的为GC_BGD，rect内部的为GC_FGD
InputOutputArray bgdModel, // 背景模型(内部使用)
InputOutputArray fgdModel, //前景模型(内部使用)
int iterCount, // 迭代次数，必须大于0
int mode = GC_EVAL // GC_INIT_WITH_RECT表示用矩形框初始化Grabcut，GC_INIT_WITH_MASK表示用掩码图像初始化Grabcut， GC_EVAL表示执行分割
);
1
2
3
4
5
6
7
8
9
10

代码：

#include <opencv2/opencv.hpp>
#include <opencv2/xfeatures2d.hpp>
#include<opencv2/face.hpp>
#include<iostream>
#include<math.h>
#include <string> 
#include<fstream> 

using namespace cv::face;
using namespace cv;
using namespace std;
using namespace cv::xfeatures2d;

int numRun = 0;//记录run了几次
Rect rect;
bool init = false;
Mat src, mask, bgmodel, fgmodel;

void showImage()//显示选择的前景区域
{
    Mat result, binMask;
    binMask.create(mask.size(), CV_8UC1);
    binMask = mask & 1;//&=操作符重载
    if (init)//init后给result设置背景色，前景色
    {
        cout << "binMask depth=" << binMask.depth() << ",type=" << binMask.type() << endl;
        src.copyTo(result, binMask);
    }
    else
    {
        src.copyTo(result);
    }
    rectangle(result, rect, Scalar(0, 0, 255), 2, 8);//绘制红色的矩形框
    imshow("src", result);
}

void setROIMask()//设置背景 前景 区域
{       // GC_FGD = 1       // 属于前景色的像素
        // GC_BGD =0;       // 属于背景色的像素
        
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40