armadillo与opencv的mat(c++版)相互转换

东方耀

armadillo与opencv的mat(c++版)相互转换

1. #include <iostream>
   
2. #include <armadillo>
   
3. #include <cmath>
   
4. #include <complex>
   
5. #include <vector>
   
6. #include "opencv2/opencv.hpp"
   
7. #include <string>
   
8. 
   
9. 
   
10. using namespace std;
    
11. 
    
12. //armadillo与opencv的mat(c++版)相互转换
    
13. 
    
14. static void Cv_mat_to_arma_mat(const cv::Mat& cv_mat_in, arma::mat& arma_mat_out)
    
15. {//convert unsigned int cv::Mat to arma::Mat<double>
    
16.     for(int r=0;r<cv_mat_in.rows;r++){
    
17.         for(int c=0;c<cv_mat_in.cols;c++){
    
18.             // 这里除以了255.0
    
19.             arma_mat_out(r,c)=cv_mat_in.data[r*cv_mat_in.cols+c]/255.0;
    
20.         }
    
21.     }
    
22. }
    
23. 
    
24. template<typename T>
    
25. static void Arma_mat_to_cv_mat(const arma::Mat<T>& arma_mat_in,cv::Mat_<T>& cv_mat_out)
    
26. {
    
27.     //通过 .memptr() 访问 Armadillo 矩阵使用的内存
    
28.     cv::transpose(cv::Mat_<T>(static_cast<int>(arma_mat_in.n_cols),
    
29.                               static_cast<int>(arma_mat_in.n_rows),
    
30.                               const_cast<T*>(arma_mat_in.memptr())),
    
31.                   cv_mat_out);
    
32. }
    
33. 
    
34. int main(){
    
35. 
    
36.     // opencv 读入一张图片 arma的矩阵 如何保存为图片呢？ 用opencv
    
37.     string img_path = "/home/jiang/图片/py_雷达算法仿真/bp成像建模与ifft_py_c++对比02.png";
    
38.    
    
39.     cv::Mat image_cv;
    
40.     image_cv = cv::imread(img_path, 0);
    
41.     cout << image_cv.channels() << endl;
    
42.     // /图像尺寸
    
43.         // cout << "size:" << image.size << endl;
    
44.         //列宽
    
45.         cout << "cols:" << image_cv.cols<<endl;
    
46.         //行高
    
47.         cout << "rows:" << image_cv.rows << endl;
    
48.         //维度
    
49.         cout << "dims:" << image_cv.dims << endl;
    
50. 
    
51. 
    
52.     //opencv的图片矩阵 转换为 arma::mat
    
53.     arma::mat img(image_cv.rows,image_cv.cols, arma::fill::zeros);
    
54.     Cv_mat_to_arma_mat(image_cv, img);   // opencv的mat 转为 arma的mat
    
55.     // img.save("img_arma.txt", arma::raw_ascii);
    
56. 
    
57.     cv::namedWindow("Display Image", cv::WINDOW_AUTOSIZE);
    
58.     cv::imshow("Display Image", image_cv);
    
59.     cv::waitKey(0);
    
60. 
    
61.     // arma::imat arma_img = arma::imat(100, 100, arma::fill::ones) * 255;
    
62.     // cv::Mat_<int> cv_img;
    
63.     // Arma_mat_to_cv_mat<int>(arma_img,cv_img);
    
64. 
    
65.     //convert arma::mat to cv::Mat
    
66.     cv::Mat_<int> cv_img;
    
67.     arma::Mat<int> arma_img = arma::Mat<int>(100, 200, arma::fill::ones) * 255;
    
68.     // umat         =         Mat<uword>
    
69. // imat         =         Mat<sword>
    
70. //sword is a typedef for a signed integer type
    
71.    
    
72.     Arma_mat_to_cv_mat<int>(arma_img,cv_img);
    
73.     cv::imwrite("b.png", cv_img);
    
74. 
    
75. 
    
76.     return 0;
    
77. }

复制代码

zouqiqi

让天下人人学会人工智能！人工智能的前景一片大好！