跟进序列化OpenCV Mat _的问题here
我的任务是我有多个序列化的OpenCV Mats。现在我想合并所有这些垫。我可以通过将这些二进制文件反序列化为Mats并使用push_back
方法合并它们来完成此操作。但是,出于我自己的原因,我必须在反序列化之前先以二进制格式合并它们。
如何合并这些二进制文件,以便最终我可以调用相同的反序列化来获取整个大型Mat?
由于
答案 0 :(得分:1)
您可以在不使用 boost 的情况下执行此操作。按照here描述的序列化方法,您可以在文件末尾附加矩阵的数据,并相应地增加最终矩阵的行数。
这是一个工作示例,matappend
完成工作。我还将matread
和matwrite
函数用于完整性:
#include <opencv2\opencv.hpp>
#include <iostream>
#include <fstream>
using namespace std;
using namespace cv;
void matwrite(const string& filename, const Mat& mat)
{
ofstream fs(filename, fstream::binary);
// Header
int type = mat.type();
int channels = mat.channels();
fs.write((char*)&mat.rows, sizeof(int)); // rows
fs.write((char*)&mat.cols, sizeof(int)); // cols
fs.write((char*)&type, sizeof(int)); // type
fs.write((char*)&channels, sizeof(int)); // channels
// Data
if (mat.isContinuous())
{
fs.write(mat.ptr<char>(0), (mat.dataend - mat.datastart));
}
else
{
int rowsz = CV_ELEM_SIZE(type) * mat.cols;
for (int r = 0; r < mat.rows; ++r)
{
fs.write(mat.ptr<char>(r), rowsz);
}
}
}
Mat matread(const string& filename)
{
ifstream fs(filename, fstream::binary);
// Header
int rows, cols, type, channels;
fs.read((char*)&rows, sizeof(int)); // rows
fs.read((char*)&cols, sizeof(int)); // cols
fs.read((char*)&type, sizeof(int)); // type
fs.read((char*)&channels, sizeof(int)); // channels
// Data
Mat mat(rows, cols, type);
fs.read((char*)mat.data, CV_ELEM_SIZE(type) * rows * cols);
return mat;
}
void matappend(const string& filename, const Mat& mat)
{
fstream fs(filename, fstream::binary | fstream::in);
// https://stackoverflow.com/a/2390938/5008845
if (fs.peek() == fstream::traits_type::eof())
{
// The file is empty, write (same as matwrite)
fs.close();
fs.open(filename, fstream::binary | fstream::out);
// Header
int type = mat.type();
int channels = mat.channels();
fs.write((char*)&mat.rows, sizeof(int)); // rows
fs.write((char*)&mat.cols, sizeof(int)); // cols
fs.write((char*)&type, sizeof(int)); // type
fs.write((char*)&channels, sizeof(int)); // channels
}
else
{
// The file is not empty, append
fs.close();
fs.open(filename, fstream::binary | fstream::out | fstream::in);
// Read Header
int rows, cols, type, channels;
fs.read((char*)&rows, sizeof(int)); // rows
fs.read((char*)&cols, sizeof(int)); // cols
fs.read((char*)&type, sizeof(int)); // type
fs.read((char*)&channels, sizeof(int)); // channels
// Consistency check
CV_Assert((cols == mat.cols) && (type == mat.type()) && (channels == mat.channels()));
// Go to beginning of file
fs.seekp(fstream::beg);
// Overwrite the number of rows
rows += mat.rows;
fs.write((char*)&rows, sizeof(int)); // rows
// Go to end of file
fs.seekp(0, fstream::end);
}
// Write data
if (mat.isContinuous())
{
fs.write(mat.ptr<char>(0), (mat.dataend - mat.datastart));
}
else
{
int rowsz = CV_ELEM_SIZE(mat.type()) * mat.cols;
for (int r = 0; r < mat.rows; ++r)
{
fs.write(mat.ptr<char>(r), rowsz);
}
}
fs.close();
}
int main()
{
// Save the random generated data
Mat1b m1 = (Mat1b(2, 2) << 1, 2, 3, 4);
Mat1b m2 = (Mat1b(3, 2) << 5, 6, 7, 8, 9, 10);
matappend("raw.bin", m1);
matappend("raw.bin", m2);
Mat m3 = matread("raw.bin");
// m3:
// 1 2
// 3 4
// 5 6
// 7 8
// 9 10
return 0;
}