上位机图像处理和嵌入式模块部署(二进制图像的读写)

【 声明:版权所有,欢迎转载,请勿用于商业用途。 联系信箱:feixiaoxing @163.com】

之前我们说过,对于图像处理而言,势必会涉及到文件的读写。但是不同格式文件的读写,这本身又是体力活,而且很多的第三方库已经帮我们实现了,比如opencv。所以,这个时候其实没有必要把精力放在文件的解析上面,然而我们在改进算法、优化性能、软硬件一体开发的同时,肯定也会涉及到文件的读写,所以这个时候建议还是用二进制文件来解决。比如说,输入到arm soc或者是mcu的时候,就会用二进制文件来代替真实的图像文件,处理完了再用python或者类似的工具恢复成bmp、或者jpeg、png图片就可以了。

1、二进制文件的格式

所谓二进制文件,就是说,文件的读取和写入都是采用二进制格式保存的。文件的前面8个字节分别代表高度和宽度。每个数据各有4个字节,且存储的时候是按照大端的方式保存的。在高度和宽度数据之后呢,就是图像数据,数据的数量就是高度乘以宽度的数值。数据中的内容,就是图像数据,一般是单字节,按照从上到下、从左到右依次保存。

复制代码
#include <iostream>
#include <fstream>
#include <vector>
#include <cstdint>
#include <Winsock2.h>

#pragma comment(lib, "Ws2_32.lib")

// image header
struct Dimensions 
{
	uint32_t height;
	uint32_t width;
};

2、写入二进制文件

写入二进制文件,一般就是把处理好的图像数据写入到文件当中去。这里面需要注意的地方,就是前面8个字节写入的时候需要进行htonl转换。后面的数据依次写入即可。

复制代码
// write binary file here
void write_binary_file(const std::string& filename, const Dimensions& dimensions, const std::vector<uint8_t>& data)
{
	std::ofstream file(filename, std::ios::binary);

	if (!file.is_open()) 
	{
		std::cerr << "Error opening file for writing." << std::endl;
		return;
	}

	// Write length and width in big-endian format
	uint32_t heightBE = htonl(dimensions.height);
	uint32_t widthBE = htonl(dimensions.width);

	file.write(reinterpret_cast<const char*>(&heightBE), sizeof(uint32_t));
	file.write(reinterpret_cast<const char*>(&widthBE), sizeof(uint32_t));

	// Write data
	file.write(reinterpret_cast<const char*>(data.data()), data.size());

	file.close();
}

3、读取二进制数据

二进制数据的读取,一般也是用于算法处理前的预操作。另外,和写入数据一样,读取数据的时候需要进行字节序的转换,即ntohl。后面的数据依次读取就可以了。

复制代码
// read binary file here
void read_binary_file(const std::string& filename, Dimensions& dimensions, std::vector<uint8_t>& data) 
{
	std::ifstream file(filename, std::ios::binary);

	if (!file.is_open()) 
	{
		std::cerr << "Error opening file for reading." << std::endl;
		return;
	}

	// Read length and width in big-endian format
	uint32_t heightBE, widthBE;

	file.read(reinterpret_cast<char*>(&heightBE), sizeof(uint32_t));
	file.read(reinterpret_cast<char*>(&widthBE), sizeof(uint32_t));

	dimensions.height = ntohl(heightBE);
	dimensions.width = ntohl(widthBE);

	// Read data
	data.resize(dimensions.height * dimensions.width);
	file.read(reinterpret_cast<char*>(data.data()), data.size());

	file.close();
}

4、代码测试

既然二进制文件写入、二进制文件读取都准备好了,那么下面就可以开始测试了。首先,我们准备高度信息和宽度信息3、4,接下来准备数据,并且将这些数据保存到bin文件当中去。保存完了,回头再用read_binary_file读取出来,如果所有的文件信息都一致,那代表没什么问题,反之就要回头检查一下原因了。当然,这里保存的数据可以是灰度图,还可以是彩色图,这个根据需求灵活进行处理,代码改起来也是很方便的。

复制代码
// file starts here
int main() 
{
	// Initialize Winsock
	WSADATA wsaData;
	if (WSAStartup(MAKEWORD(2, 2), &wsaData) != 0)
	{
		std::cerr << "WSAStartup failed." << std::endl;
		return 1;
	}

	const std::string filename = "binary_file.bin";
	Dimensions dimensions = { 3, 4 };
	std::vector<uint8_t> data = { 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12 };

	// Write to binary file
	write_binary_file(filename, dimensions, data);

	// Read from binary file
	Dimensions read_dimensions;
	std::vector<uint8_t> read_data;
	read_binary_file(filename, read_dimensions, read_data);

	std::cout << "Read length: " << read_dimensions.height << std::endl;
	std::cout << "Read width: " << read_dimensions.width << std::endl;
	std::cout << "Read data: ";

	for (const auto& value : read_data)
	{
		std::cout << static_cast<int>(value) << " ";
	}
	std::cout << std::endl;

	// Cleanup Winsock
	WSACleanup();

	return 0;
}
相关推荐
中杯可乐多加冰11 分钟前
【AI落地应用实战】AIGC赋能职场PPT汇报:从效率工具到辅助优化
人工智能·深度学习·神经网络·aigc·powerpoint·ai赋能
东临碣石8220 分钟前
【AI论文】BlenderFusion:基于三维场景的视觉编辑与生成式合成
人工智能
正在走向自律21 分钟前
第二章-AIGC入门-开启AIGC音频探索之旅:从入门到实践(6/36)
人工智能·aigc·音视频·语音识别·ai音乐·ai 音频·智能语音助手
Trent198528 分钟前
影楼精修-智能修图Agent
图像处理·人工智能·计算机视觉·aigc
烟锁池塘柳029 分钟前
【大模型】解码策略:Greedy Search、Beam Search、Top-k/Top-p、Temperature Sampling等
人工智能·深度学习·机器学习
盼小辉丶44 分钟前
PyTorch实战(14)——条件生成对抗网络(conditional GAN,cGAN)
人工智能·pytorch·生成对抗网络
Allen_LVyingbo1 小时前
数智读书笔记系列035《未来医疗:医疗4.0引领第四次医疗产业变革》
人工智能·经验分享·笔记·健康医疗
zzc9212 小时前
时频图数据集更正程序,去除坐标轴白边及调整对应的标签值
人工智能·深度学习·数据集·标签·时频图·更正·白边
isNotNullX2 小时前
什么是数据分析?常见方法全解析
大数据·数据库·数据仓库·人工智能·数据分析
riveting2 小时前
明远智睿H618:开启多场景智慧生活新时代
人工智能·嵌入式硬件·智能硬件·lga封装·3506