Simple helper method to convert from cv::Mat (32F) to Eigen::MatrixXf

Brendan Klare
1 parent 635d372c
Showing 2 changed files with 11 additions and 0 deletions
openbr/core/opencvutils.cpp
openbr/core/opencvutils.h
@@ -336,6 +336,14 @@ bool OpenCVUtils::overlaps(const QList&lt;Rect&gt; &amp;posRects, const Rect &amp;negRect, dou
     return false;
 }
+Eigen::MatrixXf OpenCVUtils::toEigen(Mat m) {
+    if (m.type() != CV_32F)
+        qFatal("Mat to Eigen Converstation only supports CV_32F");
+
+    Eigen::MatrixXf data(m.rows, m.cols);
+    return Eigen::Map<const Eigen::MatrixXf>(m.ptr<float>(), m.rows, m.cols);
+}
+
 QDataStream &operator<<(QDataStream &stream, const Mat &m)
 {
     // Write header
@@ -24,6 +24,8 @@
 #include <opencv2/core/core.hpp>
 #include <assert.h>
+#include <Eigen/Core>
+
 namespace OpenCVUtils
 {
     // Test/write/display image
@@ -91,6 +93,7 @@ namespace OpenCVUtils
     bool overlaps(const QList<cv::Rect> &posRects, const cv::Rect &negRect, double overlap);
     float overlap(const cv::Rect &rect1, const cv::Rect &rect2);
     float overlap(const QRectF &rect1, const QRectF &rect2);
+    Eigen::MatrixXf toEigen(cv::Mat m);
     int getFourcc();
 }