Opencv,在编译时确定像素的类型

Opencv, figuring out the type of a pixel at compile time

本文关键字:像素 类型 编译 Opencv      更新时间:2023-10-16

tl;dr; 如何从cv::Mat my_mat = cv::imread("myfile.png")获取编译时像素类型?

有一个我写的函数,它接受一个 Mat 迭代器:

template <typename RAI>
void my_func(RAI mat_begin, RAI mat_end) {
    typedef typename std::iterator_traits<RAI>::value_type T;
    do_stuff<T>(*mat_begin);
}
int main() {
    cv::Mat my_image = cv::imread("my_file");
    my_func(my_image.begin(), my_image.end());  
}

我可以在不明确说明RAI是什么的情况下使用此函数,因为我认为编译器能够在编译时弄清楚它是什么。

事实证明,我需要重构它以纳入整个Mat。 但是,现在我不知道如何调用do_stuff<T>,因为我不知道如何弄清楚T除了iterator_traits矩阵迭代器类型之外是什么,而且我不知道如何从我的代码中找出编译时矩阵迭代器类型,因为矩阵是用imread获得的, 这似乎在运行时设置了像素的类型,不知何故,神奇地。

OpenCV查询Mat.type()值以确定基础像素类型,然后使用 if 或 switch 语句调用特定的模板专用化。

例如,以下是floodfill.cpp中的一些片段imgproc/src/floodfill.cpp

Mat img = _image.getMat();
...
int type = img.type();
...
if( type == CV_8UC1 )
    floodFillGrad_CnIR<uchar, uchar, int, Diff8uC1>(
            img, mask, seedPoint, nv_buf.b[0], newMaskVal,
            Diff8uC1(ld_buf.b[0], ud_buf.b[0]),
            &comp, flags, &buffer);
else if( type == CV_8UC3 )
    floodFillGrad_CnIR<Vec3b, uchar, Vec3i, Diff8uC3>(
            img, mask, seedPoint, Vec3b(nv_buf.b), newMaskVal,
            Diff8uC3(ld_buf.b, ud_buf.b),
            &comp, flags, &buffer);
else if( type == CV_16UC1 )
    floodFillGrad_CnIR<unsigned short, uchar, int, Diff16uC1>(
            img, mask, seedPoint, nv_buf.s[0], newMaskVal,
            Diff16uC1(ld_buf.s[0], ud_buf.s[0]),
            &comp, flags, &buffer);
else if( type == CV_32SC1 )
    floodFillGrad_CnIR<int, uchar, int, Diff32sC1>(
            img, mask, seedPoint, nv_buf.i[0], newMaskVal,
            Diff32sC1(ld_buf.i[0], ud_buf.i[0]),
            &comp, flags, &buffer);
else if( type == CV_32SC3 )
    floodFillGrad_CnIR<Vec3i, uchar, Vec3i, Diff32sC3>(
            img, mask, seedPoint, Vec3i(nv_buf.i), newMaskVal,
            Diff32sC3(ld_buf.i, ud_buf.i),
            &comp, flags, &buffer);
else if( type == CV_32FC1 )
    floodFillGrad_CnIR<float, uchar, float, Diff32fC1>(
            img, mask, seedPoint, nv_buf.f[0], newMaskVal,
            Diff32fC1(ld_buf.f[0], ud_buf.f[0]),
            &comp, flags, &buffer);
else if( type == CV_32FC3 )
    floodFillGrad_CnIR<Vec3f, uchar, Vec3f, Diff32fC3>(
            img, mask, seedPoint, Vec3f(nv_buf.f), newMaskVal,
            Diff32fC3(ld_buf.f, ud_buf.f),
            &comp, flags, &buffer);
else
    CV_Error(CV_StsUnsupportedFormat, "");