• cv.Mat 与 .txt 文件数据的读写操作


    1、按OpenCV格式实现的 .txt 文件读写

    可以用 cvSave 和 cvLoad 实现,格式和 .xml/.yml 的差不多,不过如果专用与 OpenCV 的数据读写,还是用  .xml/.yml 文件格式较好,我比较喜欢 .yml 格式,可读性非常棒。

    用 cvSave 和 cvLoad 读写 .txt 文件,其实现方式和数据格式与 .yml 文件基本一致。

    例如:cvSave("camera_matrix.txt",camera_matrix);  //保存了 camera_matrix 的数组头以及和它所指的data(类似yml形式的文件)

    2、导入/导出其它程序的 .txt 文件数据

    可以用常规的 sprintf_s 和 fprintf_s 来实现,不过效率比较低,这里介绍一种快捷易用的方法,利用了 std 的 steam 和 vector。

      1 #include <iostream>
      2 #include <fstream>
      3 #include <iterator>
      4 #include <vector>
      5 
      6 using namespace std;
      7 
      8 /*----------------------------
      9  * 功能 : 将 cv::Mat 数据写入到 .txt 文件
     10  *----------------------------
     11  * 函数 : WriteData
     12  * 访问 : public 
     13  * 返回 : -1:打开文件失败;0:写入数据成功;1:矩阵为空
     14  *
     15  * 参数 : fileName    [in]    文件名
     16  * 参数 : matData    [in]    矩阵数据
     17  */
     18 int WriteData(string fileName, cv::Mat& matData)
     19 {
     20     int retVal = 0;
     21 
     22     // 检查矩阵是否为空
     23     if (matData.empty())
     24     {
     25         cout << "矩阵为空" << endl; 
     26         retVal = 1;
     27         return (retVal);
     28     }
     29 
     30     // 打开文件
     31     ofstream outFile(fileName.c_str(), ios_base::out);    //按新建或覆盖方式写入
     32     if (!outFile.is_open())
     33     {
     34         cout << "打开文件失败" << endl; 
     35         retVal = -1;
     36         return (retVal);
     37     }
     38 
     39     // 写入数据
     40     for (int r = 0; r < matData.rows; r++)
     41     {
     42         for (int c = 0; c < matData.cols; c++)
     43         {
     44             int data = matData.at<uchar>(r,c);    //读取数据,at<type> - type 是矩阵元素的具体数据格式
     45             outFile << data << "	" ;    //每列数据用 tab 隔开
     46         }
     47         outFile << endl;    //换行
     48     }
     49 
     50     return (retVal);
     51 }
     52 
     53 
     54 /*----------------------------
     55  * 功能 : 从 .txt 文件中读入数据,保存到 cv::Mat 矩阵
     56  *        - 默认按 float 格式读入数据,
     57  *        - 如果没有指定矩阵的行、列和通道数,则输出的矩阵是单通道、N 行 1 列的
     58  *----------------------------
     59  * 函数 : LoadData
     60  * 访问 : public 
     61  * 返回 : -1:打开文件失败;0:按设定的矩阵参数读取数据成功;1:按默认的矩阵参数读取数据
     62  *
     63  * 参数 : fileName    [in]    文件名
     64  * 参数 : matData    [out]    矩阵数据
     65  * 参数 : matRows    [in]    矩阵行数,默认为 0
     66  * 参数 : matCols    [in]    矩阵列数,默认为 0
     67  * 参数 : matChns    [in]    矩阵通道数,默认为 0
     68  */
     69 int LoadData(string fileName, cv::Mat& matData, int matRows = 0, int matCols = 0, int matChns = 0)
     70 {
     71     int retVal = 0;
     72 
     73     // 打开文件
     74     ifstream inFile(fileName.c_str(), ios_base::in);
     75     if(!inFile.is_open())
     76     {
     77         cout << "读取文件失败" << endl;
     78         retVal = -1;
     79         return (retVal);
     80     }
     81 
     82     // 载入数据
     83     istream_iterator<float> begin(inFile);    //按 float 格式取文件数据流的起始指针
     84     istream_iterator<float> end;            //取文件流的终止位置
     85     vector<float> inData(begin,end);        //将文件数据保存至 std::vector 中
     86     cv::Mat tmpMat = cv::Mat(inData);        //将数据由 std::vector 转换为 cv::Mat
     87 
     88     // 输出到命令行窗口
     89     //copy(vec.begin(),vec.end(),ostream_iterator<double>(cout,"	")); 
     90 
     91     // 检查设定的矩阵尺寸和通道数
     92     size_t dataLength = inData.size();
     93     //1.通道数
     94     if (matChns == 0)
     95     {
     96         matChns = 1;
     97     }
     98     //2.行列数
     99     if (matRows != 0 && matCols == 0)
    100     {
    101         matCols = dataLength / matChns / matRows;
    102     } 
    103     else if (matCols != 0 && matRows == 0)
    104     {
    105         matRows = dataLength / matChns / matCols;
    106     }
    107     else if (matCols == 0 && matRows == 0)
    108     {
    109         matRows = dataLength / matChns;
    110         matCols = 1;
    111     }
    112     //3.数据总长度
    113     if (dataLength != (matRows * matCols * matChns))
    114     {
    115         cout << "读入的数据长度 不满足 设定的矩阵尺寸与通道数要求,将按默认方式输出矩阵!" << endl;
    116         retVal = 1;
    117         matChns = 1;
    118         matRows = dataLength;
    119     } 
    120 
    121     // 将文件数据保存至输出矩阵
    122     matData = tmpMat.reshape(matChns, matRows).clone();
    123     
    124     return (retVal);
    125 }
  • 相关阅读:
    学姐学长们的测试
    loli的测试——搜索
    搜索-2
    单调队列
    乱搞题 (不知道怎么分类)
    对拍及数据生成
    搜索-1
    矩形重叠
    拼写单词
    java笔记
  • 原文地址:https://www.cnblogs.com/ttzm/p/5997796.html
Copyright © 2020-2023  润新知