c++代码 将yolov8预测出的mask图还原到原图大小

以下是一个示例 C++ 代码，用于将 YOLOv8 预测出的 mask 图像还原为原始图像大小。请注意，此代码仅适用于将 mask 图像转换为二值图像并将其与原始图像进行按位与运算的情况：

#include <opencv2/opencv.hpp>

using namespace cv;

// 还原 mask 图像到原始图像大小
Mat restoreMask(Mat mask, Mat original) {
    // 缩放 mask 图像到原始图像大小
    Mat resizedMask;
    resize(mask, resizedMask, original.size(), 0, 0, INTER_NEAREST);

    // 将 mask 图像二值化
    Mat binaryMask;
    threshold(resizedMask, binaryMask, 0, 255, THRESH_BINARY);

    // 将二值化后的 mask 图像与原始图像进行按位与运算
    Mat restored;
    bitwise_and(original, binaryMask, restored);

    return restored;
}

int main() {
    // 读取原始图像和 mask 图像
    Mat original = imread("original.jpg");
    Mat mask = imread("mask.jpg", IMREAD_GRAYSCALE);

    // 还原 mask 图像到原始图像大小
    Mat restored = restoreMask(mask, original);

    // 显示原始图像和还原后的图像
    imshow("Original", original);
    imshow("Restored", restored);
    waitKey(0);

    return 0;
}

在此代码中，`restoreMask()` 函数接受两个输入参数：`mask` 是预测出的 mask 图像，`original` 是原始图像。该函数首先将 mask 图像缩放到原始图像大小，然后将其二值化，并将其与原始图像进行按位与运算，得到还原后的图像。最后，该函数返回还原后的图像。

在主函数中，我们读取原始图像和 mask 图像，并通过 `restoreMask()` 函数将 mask 图像还原为原始图像大小。最后，我们显示原始图像和还原后的图像。请注意，此代码仅适用于将 mask 图像转换为二值图像并将其与原始图像进行按位与运算的情况，您需要根据实际情况进行修改。