c++ - 编写接受 cv::Mat 或 cv::UMat 类型输入的模板函数
问题描述
编写接受 Mat 或 UMat 类型输入参数的函数的最简单和最干净的方法是什么?
我应该使用 InputArray,使用模板还是有更好的选择?我现在拥有为 Mat 和 UMat 编写的具有相同实现的功能。
该函数应充分利用 OpenCL 上的 UMat 抽象,并且运行速度大致与仅为 UMats 编写的一样快,并且无需将 UMats 复制到 Mats 的开销。
下面是我想为 Mat 和 UMat 定义的函数示例(请不要建议重构以删除本地 Mat/UMat 变量,这只是一个示例)
using namespace cv;
void foo(const Mat & in1, const Mat & in2, Mat & out)
{
Mat aux1;
Mat aux2;
exp(in1, aux1);
exp(in2, aux2);
add(aux1, aux2, out);
return;
}
void foo(const UMat & in1, const UMat & in2, UMat & out)
{
UMat aux1;
UMat aux2;
exp(in1, aux1);
exp(in2, aux2);
add(aux1, aux2, out);
return;
}
解决方案
由@Miki 和@Gianni 建议,我们可以选择InpputArray
和OutputArray
的通用模板类型 Mat, Mat_<T>, Matx<T, m, n>, std::vector<T>, std::vector<std::vector<T> >, std::vector<Mat>, std::vector<Mat_<T> >, UMat, std::vector<UMat> or double.
void func(InputArray _src1, InputArray _src2, OutputArray _dst)
{
Mat src1 = _src1.getMat(), src2 = _src2.getMat();
CV_Assert( src1.type() == src2.type() && src1.size() == src2.size());
Mat aux1 = Mat(src1.size(), src1.type());
Mat aux2 = Mat(src1.size(), src1.type());
cv::exp(src1, aux1);
cv::exp(src2, aux2);
_dst.create(src1.size(), src1.type());
Mat dst = _dst.getMat();
cv::add(aux1, aux2, dst);
}
现在你可以用Mat
orUMat
甚至std::vector
int test(){
std::vector<float> vec1 = {1,2,3};
std::vector<float> vec2 = {3,2,1};
std::vector<float> dst;
func(vec1, vec2, dst);
// now dst is [22.8038, 14.7781, 22.8038]
}
推荐阅读
- php - PHP shell_exec 复合命令在后台运行
- d3.js - 如何在数据更新期间同步散点和折线图?
- svelte - 在浏览器中更改选项卡时访问存储在 Svelte 存储中的数据
- python - 去除轮廓区域外的图片
- java - 如何将外部 keylistener 类调用到项目文本字段中?
- firebase - Cloud Firestore 查询以获取文档 ID,Flutter
- apache - 从凌晨 12:30 到凌晨 2 点阻止代理用户
- python - 如何在 {{ form.label }} 中输入默认值
- python - 在 python-telegram-bot 中每 15 分钟发送一次消息
- django - 我可以在 2 个场景中使用一个 Django Modelform 并使用不同数量的字段吗?