cvtColor封装

原创已于 2022-03-18 19:53:28 修改 · 1.1k 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#opencv

于 2022-03-18 19:52:36 首次发布

【opencv学习】专栏收录该内容

68 篇文章

订阅专栏

本文档介绍了一个关于OpenCV中cvtColor函数报错的问题，作者提供了一个自定义的函数cvtColor_src来解决这个问题。该函数根据输入图像的通道数，智能选择合适的颜色空间转换方式，包括4通道、3通道、2通道和1通道的图像。对于错误的图像通道数，函数会直接返回原图像。这个封装有助于简化代码并提高鲁棒性。

cvtColor(src, gray);报错，改成如下函数进行封装

void cvtColor_src(Mat &src, Mat &src_gray)
{
    //  转换4通道
    if (src.channels() == 4) {
        cv::cvtColor(src, src_gray, CV_BGRA2GRAY);
    }
    //  转换3通道
    else if (src.channels() == 3) {
        cv::cvtColor(src, src_gray, CV_BGR2GRAY);
    }
    //  转换2通道
    else if (src.channels() == 2) {
        cv::cvtColor(src, src_gray, CV_BGR5652GRAY);
    }
    //  转换单通道
    else if (src.channels() == 1) {// 单通道的图片直接就不需要处理
        src_gray = src;
    }
    else { // 负数,说明图有问题 直接返回   
        src_gray = src;
    }
 
}