c++ - 使用 OpenCV 保存整数 CV_32S 图像

Question

我正在处理包含有符号整数数据的 TIF 图像。成功输入并处理后，我需要以相同的格式输出图像(输入和输出两个 *.tif 文件)。

对于输入，我知道OpenCV 不知道数据是有符号的还是无符号的，所以它假设是无符号的。使用 this trick解决了这个问题(手动切换 cv::Mat 的类型)。

但是，当我输出图像并再次加载它时，我没有得到预期的结果。该文件包含多个段(像素组)，格式如下(我必须使用此格式):

• 不属于任何段的所有像素的值为 -9999
• 属于单个段的所有像素具有相同的正整数值
• (例如，第一段的所有像素都具有值 1，第二段的所有像素都具有 2 等)

下面是示例代码:

void ImageProcessor::saveSegments(const std::string &filename){
    cv::Mat segmentation = cv::Mat(workingImage.size().height,
                                   workingImage.size().width,
                                   CV_32S, cv::Scalar(-9999));

    for (int i=0, szi = segmentsInput.size(); i < szi; ++i){
        for (int j=0, szj = segmentsInput[i].size(); j < szj; ++j){
            segmentation.at<int>(segmentsInput[i][j].Y,
                                 ssegmentsInput[i][j].X) = i+1;
        }
    }
    cv::imwrite(filename, segmentation);
}

您可以假设所有变量(例如 workingImage、segmentsInput)都作为全局变量存在。

使用此代码，当我输入图像并检查值时，大多数值都设置为 0 而设置的值采用完整范围的整数值(在我的示例中我有 20 个片段)。

Answer 1

您不能直接使用 imwrite 保存整数矩阵。作为the documentation状态:“只有 8 位(或 16 位无符号 (CV_16U) 在 PNG、JPEG 2000 和 TIFF 的情况下)单 channel 或 3 channel (具有 'BGR' channel 顺序)图像可以使用此功能保存。 "

但是，您可以将 CV_32S 矩阵转换为 CV_8UC4 并将其另存为不压缩的 PNG。当然，这有点不安全，因为 endianness开始发挥作用，可能会在不同的系统或编译器之间更改您的值(特别是因为我们在这里谈论的是有符号整数)。如果你总是使用相同的系统和编译器，你可以使用这个:

cv::Mat segmentation = cv::Mat(workingImage.size().height,
                               workingImage.size().width,
                               CV_32S, cv::Scalar(-9999));
cv::Mat pngSegmentation(segmentation.rows, segmentation.cols, CV_8UC4, (cv::Vec4b*)segmentation.data);
std::vector<int> params;
params.push_back(CV_IMWRITE_PNG_COMPRESSION);
params.push_back(0);
cv::imwrite("segmentation.png", pngSegmentation, params);