利用fstream,将ncnn及opencv的mat存储成bin文件。
ncnn::Mat to bin
std::ios::binary标志指示文件以二进制模式进行读写,
cpp
std::ofstream file("output_x86.bin", std::ios::binary);
将input_mat中的宽、高和通道数分别赋值给width、height和channels,
cpp
int width = input_mat.w;
int height = input_mat.h;
int channels = input_mat.c;
使用file.write函数将width、height和channels的值以二进制形式写入文件。reinterpret_cast用于将指针类型进行类型转换,这里将 int* 类型转换为 const char* 类型,以便将其作为字节流写入文件,
cpp
file.write(reinterpret_cast<const char*>(&width), sizeof(int));
file.write(reinterpret_cast<const char*>(&height), sizeof(int));
file.write(reinterpret_cast<const char*>(&channels), sizeof(int));
将input_mat.data所指向的数据以二进制形式写入文件。input_mat.data是一个指向input_mat数据的指针,reinterpret_cast将其转换为const char*类型,以便将其作为字节流写入文件。input_mat.cstep * sizeof(float)表示要写入的字节数,input_mat.cstep是input_mat数据中每一行的字节数,乘以 sizeof(float) 表示将所有数据写入文件,
cpp
file.write(reinterpret_cast<const char*>(input_mat.data), input_mat.cstep * sizeof(float));
file.close();
cv::Mat to bin
cpp
std::ofstream file(filename, std::ios::binary);
int width = mat.cols;
int height = mat.rows;
int channels = mat.channels();
file.write(reinterpret_cast<const char*>(&width), sizeof(int));
file.write(reinterpret_cast<const char*>(&height), sizeof(int));
file.write(reinterpret_cast<const char*>(&channels), sizeof(int));
file.write(reinterpret_cast<const char*>(mat.data), mat.total() * mat.elemSize());
file.close();