可见光图像转红外图像网络设计
1、立项背景和目标:
本项目旨在解决夜间安防、自动驾驶等领域中,红外图像采集硬件成本高昂、且在部分场景难以部署的问题。目标是设计并训练一个深度学习网络,在仅有可见光摄像头的情况下,通过算法将可见光图像转换为具有热辐射特征的红外图像,从而降低硬件成本,实现跨模态数据增强。
2、软件功能、核心功能模块介绍:
核心功能是基于编码器-解码器架构的图像转换模型。包含三大模块:数据预处理模块(统一256x256尺寸与归一化)、生成器网络模块(6层下采样提取特征,6层上采样重建图像,并引入跳跃连接融合多尺度信息)、以及图像输出模块(使用OpenCV进行后处理与保存)。
3、业务流程、功能路径描述:
用户输入一张可见光图像 -> 系统自动进行尺寸缩放和像素归一化 -> 载入训练好的PyTorch模型权重,进入评估模式 -> 输入数据经过生成器前向传播 -> 输出张量转换为NumPy数组 -> 映射回0-255像素区间 -> 最终输出高质量的红外模拟图像。
大数据