#include <iostream> #include <fstream> #include <iterator> #include <vector>  using namespace std;  /*----------------------------  * 功能 : 將 cv::Mat 數據寫入到 .txt 文件  *----------------------------  * 函數 : WriteData  * 訪問 : public  * 返回 : -1：打開文件失敗；0：寫入數據成功；1：矩陣為空  *  * 參數 : fileName [in] 文件名  * 參數 : matData [in] 矩陣數據  */ int WriteData(string fileName, cv::Mat& matData) {  int retVal = 0;   // 打開文件  ofstream outFile(fileName.c_str(), ios_base::out); //按新建或覆蓋方式寫入  if (!outFile.is_open())  {  cout << "打開文件失敗" << endl;  retVal = -1;  return (retVal);  }   // 檢查矩陣是否為空  if (matData.empty())  {  cout << "矩陣為空" << endl;  retVal = 1;  return (retVal);  }   // 寫入數據  for (int r = 0; r < matData.rows; r++)  {  for (int c = 0; c < matData.cols; c++)  {  uchar data = matData.at<uchar>(r,c); //讀取數據，at<type> - type 是矩陣元素的具體數據格式  outFile << data << "/t" ; //每列數據用 tab 隔開  }  outFile << endl; //換行  }   return (retVal); }   /*----------------------------  * 功能 : 從 .txt 文件中讀入數據，保存到 cv::Mat 矩陣  * - 默認按 float 格式讀入數據，  * - 如果沒有指定矩陣的行、列和通道數，則輸出的矩陣是單通道、N 行 1 列的  *----------------------------  * 函數 : LoadData  * 訪問 : public  * 返回 : -1：打開文件失敗；0：按設定的矩陣參數讀取數據成功；1：按默認的矩陣參數讀取數據  *  * 參數 : fileName [in] 文件名  * 參數 : matData [out] 矩陣數據  * 參數 : matRows [in] 矩陣行數，默認為 0  * 參數 : matCols [in] 矩陣列數，默認為 0  * 參數 : matChns [in] 矩陣通道數，默認為 0  */ int LoadData(string fileName, cv::Mat& matData, int matRows = 0, int matCols = 0, int matChns = 0) {  int retVal = 0;   // 打開文件  ifstream inFile(fileName.c_str(), ios_base::in);  if(!inFile.is_open())  {  cout << "讀取文件失敗" << endl;  retVal = -1;  return (retVal);  }   // 載入數據  istream_iterator<float> begin(inFile); //按 float 格式取文件數據流的起始指針  istream_iterator<float> end; //取文件流的終止位置  vector<float> inData(begin,end); //將文件數據保存至 std::vector 中  cv::Mat tmpMat = cv::Mat(inData); //將數據由 std::vector 轉換為 cv::Mat   // 輸出到命令行窗口  //copy(vec.begin(),vec.end(),ostream_iterator<double>(cout,"/t"));   // 檢查設定的矩陣尺寸和通道數  size_t dataLength = inData.size();  //1.通道數  if (matChns == 0)  {  matChns = 1;  }  //2.行列數  if (matRows != 0 && matCols == 0)  {  matCols = dataLength / matChns / matRows;  }  else if (matCols != 0 && matRows == 0)  {  matRows = dataLength / matChns / matCols;  }  else if (matCols == 0 && matRows == 0)  {  matRows = dataLength / matChns;  matCols = 1;  }  //3.數據總長度  if (dataLength != (matRows * matCols * matChns))  {  cout << "讀入的數據長度 不滿足 設定的矩陣尺寸與通道數要求，將按默認方式輸出矩陣！" << endl;  retVal = 1;  matChns = 1;  matRows = dataLength;  }   // 將文件數據保存至輸出矩陣  matData = tmpMat.reshape(matChns, matRows).clone();   return (retVal); }