【opencv】示例-grabcut.cpp 使用OpenCV库的GrabCut算法进行图像分割

left mouse button - set rectangle

SHIFT+left mouse button - set GC_FGD pixels

CTRL+left mouse button - set GC_BGD pixels

这段代码是一个使用OpenCV库的GrabCut算法进行图像分割的C++程序。它允许用户通过交互式方式选择图像中的一个区域，并利用GrabCut算法尝试将其分割出来。代码中包含用户操作指南、颜色定义、事件处理以及GrabCut算法的迭代过程，实现了用户通过鼠标交互和按键控制来迭代分割效果。程序逻辑结构清晰，采用面向对象的方式封装了GrabCut分割相关的操作。

// 引入OpenCV库中处理图像编解码的相关头文件
#include "opencv2/imgcodecs.hpp"
// 引入OpenCV库中高级GUI的相关头文件
#include "opencv2/highgui.hpp"
// 引入OpenCV库中图像处理的相关头文件
#include "opencv2/imgproc.hpp"


// 引入输入输出流的库文件
#include <iostream>


// 使用标准命名空间std，免去std::前缀
using namespace std;
// 使用OpenCV命名空间cv，免去cv::前缀
using namespace cv;


// 定义静态辅助函数，用于显示程序帮助信息
static void help(char** argv)
{
    cout << "\nThis program demonstrates GrabCut segmentation -- select an object in a region\n"
            "and then grabcut will attempt to segment it out.\n"
            "Call:\n"
        << argv[0] << " <image_name>\n" // 显示如何调用程序
        "\nSelect a rectangular area around the object you want to segment\n"
        // 以下是热键操作说明
        "\nHot keys: \n"
        "\tESC - quit the program\n" // 按ESC键退出程序
        "\tr - restore the original image\n" // 按'r'键还原到原始图像
        "\tn - next iteration\n" // 按'n'键执行下一迭代步骤
        "\n"
        // 以下是鼠标操作说明
        "\tleft mouse button - set rectangle\n" // 左键点击设置矩形区域
        "\n"
        "\tCTRL+left mouse button - set GC_BGD pixels\n" // 按住CTRL键同时左键点击设置背景像素
        "\tSHIFT+left mouse button - set GC_FGD pixels\n" // 按住SHIFT键同时左键点击设置前景像素
        "\n"
        "\tCTRL+right mouse button - set GC_PR_BGD pixels\n" // 按住CTRL键同时右键点击设置可能的背景像素
        "\tSHIFT+right mouse button - set GC_PR_FGD pixels\n" << endl; // 按住SHIFT键同时右键点击设置可能的前景像素
}


// 定义一些标量常量表示不同的颜色
const Scalar RED = Scalar(0,0,255);
const Scalar PINK = Scalar(230,130,255);
const Scalar BLUE = Scalar(255,0,0);
const Scalar LIGHTBLUE = Scalar(255,255,160);
const Scalar GREEN = Scalar(0,255,0);


// 定义背景键和前景键的标志，用于鼠标操作事件中
const int BGD_KEY = EVENT_FLAG_CTRLKEY;
const int FGD_KEY = EVENT_FLAG_SHIFTKEY;


// 声明静态函数，用于从复合掩模图中提取二值掩模图
static void getBinMask( const Mat& comMask, Mat& binMask )
{
    // 确保comMask非空且类型为CV