Opencv,在编译时计算出像素的类型

时间:2017-10-27 19:55:34

标签: c++ opencv matrix

TL;博士;如何从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得到的,它似乎在运行时设置像素的类型不知何故,神奇地。

1 个答案:

答案 0 :(得分:1)

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

例如,以下是floodfill.cppimgproc/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, "");