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
得到的,它似乎在运行时设置像素的类型不知何故,神奇地。
答案 0 :(得分:1)
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, "");