调用opencv grabcut写一个C++程序 包括源文件和头文件

以下是一个使用 OpenCV 实现 grabcut 的 C++ 程序，包括源文件和头文件：

源文件 `grabcut.cpp`：

#include <opencv2/opencv.hpp>

int main()
{
    // 读取输入图像
    cv::Mat image = cv::imread("input.jpg");

    // 定义前景和背景的掩码，0 表示背景，1 表示前景，2 表示未知区域
    cv::Mat mask(image.size(), CV_8UC1, cv::Scalar(2));

    // 定义矩形框，指定前景区域
    cv::Rect rect(100, 100, 200, 200);
    cv::Mat bgdModel, fgdModel;

    // 进行 grabcut 分割
    cv::grabCut(image, mask, rect, bgdModel, fgdModel, 5, cv::GC_INIT_WITH_RECT);

    // 将未知区域设为背景
    mask.setTo(cv::Scalar(0), mask == cv::GC_PR_FGD);
    mask.setTo(cv::Scalar(1), mask == cv::GC_FGD);

    // 将前景提取出来
    cv::Mat foreground(image.size(), CV_8UC3, cv::Scalar(0, 0, 0));
    image.copyTo(foreground, mask);

    // 保存结果
    cv::imwrite("output.jpg", foreground);

    return 0;
}

头文件 `grabcut.h`：

#ifndef GRABCUT_H
#define GRABCUT_H

#include <opencv2/opencv.hpp>

void grabcut(const cv::Mat& image, cv::Mat& mask, const cv::Rect& rect);

#endif

在这个程序中，我们使用了 OpenCV 提供的 `cv::grabCut` 函数进行 grabcut 分割，并将结果保存到输出图像中。头文件中也提供了一个 `grabcut` 函数，可以在其他程序中调用。