这是我的Eigen+OpenCV互操作示例,希望对你有用:
#define EIGEN_RUNTIME_NO_MALLOC
#include <Eigen/Dense>
#include <Eigen/Sparse>
#include <vector>
#include <Eigen/IterativeLinearSolvers>
#include <iostream>
#include "opencv2/core/eigen.hpp"
#include "opencv2/opencv.hpp"
using namespace Eigen;
using namespace cv;
using namespace std;
void EnergyFilter(Mat& src,Mat& dst,double alpha)
{
int n_pixels=src.rows*src.cols;
Mat m=src.reshape(1,n_pixels).clone();
m.convertTo(m,CV_64FC1);
VectorXd I(n_pixels);
VectorXd u(n_pixels);
cv2eigen(m,I);
SparseMatrix<double> A(n_pixels,n_pixels);
typedef Eigen::Triplet<double> T;
std::vector<T> tripletList;
for(int i=0;i<n_pixels;i++)
{
tripletList.push_back(T(i,i,1+4*alpha));
if((i+1) < n_pixels){tripletList.push_back(T(i,i+1,-alpha));}
if((i-1) >= 0){tripletList.push_back(T(i,i-1,-alpha));}
if((i+src.cols) < n_pixels){tripletList.push_back(T(i,i+src.cols,-alpha));}
if((i-src.cols) >= 0){tripletList.push_back(T(i,i-src.cols,-alpha));}
}
tripletList.push_back(T(0,0,1+2*alpha));
for(int i=1;i<src.cols;i++)
{
tripletList.push_back(T(i,i,1+3*alpha));
}
tripletList.push_back(T(n_pixels-1,n_pixels-1,1+2*alpha));
for(int i=1;i<src.cols;i++)
{
tripletList.push_back(T(i,n_pixels-i-1,1+3*alpha));
}
A.setFromTriplets(tripletList.begin(),tripletList.end());
tripletList.clear();
ConjugateGradient<SparseMatrix<double> > cg;
cg.compute(A);
u = cg.solve(I);
std::cout << "#iterations: " << cg.iterations() << std::endl;
std::cout << "estimated error: " << cg.error() << std::endl;
dst=Mat(n_pixels,1,CV_64FC1);
eigen2cv(u,dst);
dst=dst.reshape(1,src.rows);
dst.convertTo(dst,CV_8UC1);
}
int main(int argc, char* argv[])
{
namedWindow("image");
namedWindow("result");
Mat img=imread("d:\\ImagesForTest\\lena.jpg",1);
imshow("image",img);
waitKey(10);
Mat res;
vector<Mat> ch;
cv::split(img,ch);
for(int i=0;i<3;i++)
{
EnergyFilter(ch[i],res,3);
res.copyTo(ch[i]);
}
cv::merge(ch,res);
imshow("result",res);
waitKey(0);
return 0;
}