Eigen 与 OpenCV 数据转换

Yiwei Zhang · 2023-01-13 · via 又见苍岚

OpenCV 支持与 Eigen 之间的数据转化，本文记录方法。

数据转换

OpenCV 算子

cv -> eigen: cv2eigen()
eigen -> cv: eigen2cv()
需要引入：
1
#include <opencv2/core/eigen.hpp>
官方文档：https://docs.opencv.org/4.5.5/d0/daf/group__core__eigen.html#ga1add06b744a69bc05e1e16a5eb20be3e

示例

转换代码

#include <iostream>
#include <Eigen/Core>
#include <unsupported/Eigen/CXX11/Tensor>
#include <opencv2/core/eigen.hpp>
#include <opencv2/opencv.hpp>using namespace Eigen;
using namespace std;
using namespace cv;
int main()
{
    Mat img = imread("test.jpg");
    Mat img2;
    Eigen::Tensor<uint8_t, 3> a_tensor;
    cv2eigen(img, a_tensor);
    a_tensor.slice(Eigen::array<size_t, 3> {500, 500, 0}, Eigen::array<size_t, 3> {200, 400, 3}).setZero();
    eigen2cv(a_tensor, img2);
}

效果示例，可以看到 Tensor 按照预定想法处理了图像

Eigen 转 OpenCV 时需要 Tensor 为 3 维矩阵，不然会报错：

如果是二维 Tensor 需要升维

Eigen::array<int, 3> three_dims{ {H, W, 1} };
Tensor<float, 3> result_map_3d = result_map_tmp_2d.reshape(three_dims);Mat res;
eigen2cv(result_map_3d, res);

与 OpenCV 官网示例代码有所出入，建议不加 Eigen::RowMajor。

说明：这里说明一下，OpenCV 官网文档建议这样建立 Tensor 对象：Eigen::Tensor<uint8_t, 3, Eigen::RowMajor> a_tensor;

但是这样建立对象后简单的操作就会报错，建议不加 Eigen::RowMajor，截至目前我在代码中没发现删除这句话的弊端。

参考资料

文章链接：
https://www.zywvvd.com/notes/coding/cpp/eigen/eigen-opencv/

此内容由惯性聚合(RSS阅读器)自动聚合整理，仅供阅读参考。原文来自 — 版权归原作者所有。

推荐订阅源

又见苍岚

数据转换

OpenCV 算子

示例

参考资料