一种基于改进轻量化YOLOv8模型的夜间红外检测方法及系统

本发明提出一种基于改进轻量化YOLOv8模型的夜间红外检测方法及系统，涉及深度学习、目标检测的技术领域。首先，获取红外图像数据集，对红外图像数据集进行预处理，并将预处理后的红外图像数据集划分为训练集、验证集和测试集；构建改进轻量化YOLOv8模型；所述改进轻量化YOLOv8模型在YOLOv8模型的基础上加入用于提高红外图像多尺度及特征提取能力的DWR模块、用于提高图像特征融合能力的CGAFusion模块和LSDECD_Detect模块；利用训练集对改进轻量化YOLOv8模型进行训练，并在训练过程中，利用验证集验证改进轻量化YOLOv8模型的有效性，得到训练好的改进轻量化YOLOv8模型；将测试集输入至训练好的改进轻量化YOLOv8模型，得到红外检测结果，实现在夜间能见度较差的情况下，提高车辆检测准确率。