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