c++ - 如何在 C++ 中读取文本文件并创建 Mat 对象

Question

嗨，我已经能够将 Mat 对象写入文本文件。如下，

std::fstream outputFile;
    outputFile.open( "myFile.txt", std::ios::out ) ;

    outputFile << des_object.rows << std::endl;
    outputFile << des_object.cols << std::endl;

    for(int i=0; i<des_object.rows; i++)
    {
        for(int j=0; j<des_object.cols; j++)
        {
            outputFile << des_object.at<float>(i,j) << std::endl;
        }

    }
    outputFile.close( );

在我的前两行代码中，我打印了行数和列数，以便在我读回它时使用。但我无法读取文本文件并再次创建 Mat 对象。

以下是我尝试过的代码。不确定我的代码是否正确。

Mat des_object1;
    std::ifstream file("myFile.txt");
    std::string str; 
    int rows;
    int cols;
    int a = 0;
    while (std::getline(file, str))
    {
        int i = 0;
        int j = 0;

        if(a == 0){
            rows = std::stoi( str );
        }else if(a == 1){
            cols = std::stoi( str );
        }else{

            for(i; i< rows; i++)
            {
                for(j; j<cols; j++)
                {
                     des_object1.at<float>(i,j) = ::atof(str.c_str());
                     break;
                }
            }

        }
        ++a;
    }

Answer 1

使用 opencv FileStorage可能要容易得多：

// write:
Mat m;
FileStorage fs("myfile.txt",FileStorage::WRITE);
fs << "mat1" << m;

// read:
FileStorage fs("myfile.txt",FileStorage::READ);
fs["mat1"] >> m;

Answer 2

您的内部循环位于错误的位置：每次迭代读取一行时，您都需要保留计数器。不过，它可以做得更简单：

if (in >> rows >> cols) {
    // resize the matrix to its proper size
    for (int r(0); r!= rows; ++r) {
        for (int c(0); c != cols; ++c) {
            if (!(in >> mat[r][c])) {
                throw std:: runtime_error("failed to read matrix");
            }
        }
    }
}

Answer 3

尝试这样的事情：

while在循环外初始化这两个

int k=0;
int l=0;

而不是使用for循环

if(j<cols){
    des_object1.at<float>(k,l) = ::atof(str.c_str());
}else{
    k=0;
    l++;
    des_object1.at<float>(k,l) = ::atof(str.c_str());
}
j++;

Answer 4

我写了一个方法来读取一个Mat也.asc应该适用的方法.txt。

也许有更时尚、更有效的方法可以做到这一点，但这种方法有效且易于理解。

头

int Load_From_Path_Text(Mat *pMA_Out, string path)

变量

ifstream        IS_File;
string          ST_Line;
stringstream    SS_Line;
unsigned int    rows        = 0;
unsigned int    cols        = 0;
unsigned int    y           = 0;
unsigned int    x           = 0;
float           F_Value;

获取大小

IS_File.open(path);
while(getline(IS_File, ST_Line))
{
    if(cols == 0)
    {
        SS_Line << ST_Line;
        while(SS_Line >> F_Value)
            cols++;
    }

    rows++;
}
IS_File.close();

创建图像

*pMA_Out = Mat(rows, cols, CV_32FC1);

读取数据

IS_File.open(path);
while(getline(IS_File, ST_Line))
{
    SS_Line.clear();
    SS_Line << ST_Line;
    x = 0;

    while(SS_Line >> F_Value)
    {
        pMA_Out->at<float>(y, x) = F_Value;
        x++;
    }

    y++;
}
IS_File.close();

例子

1 1 1 -2
1 2 1 0.5
1 1 3 2.1
1 1 1 1.5

转向 这个 （在beeing转换为CV_8UC1使用normalizewithCV_MINMAX显示后）。