基于红外图像的地铁漏水识别方法

本发明涉及红外成像技术，具体提供了基于红外图像的地铁漏水识别方法，包括模型优化，模型优化包括以下步骤：（A1）采集地铁渗漏水的红外图像，从而构建数据集；（A2）对数据集进行预处理：去除噪声，并增强漏水部分；（A3）对预处理后的图像进行标注，标注出漏水区域的边缘，将标注结果转换为YOLO格式的数据集，并划分出训练集、验证集和测试集；（A4）在目标检测任务中，将小目标的定位损失Lloc修正为Lˊloc，获得优化后的YOLOv8模型；（A5）使用训练集对优化后的模型进行训练，使用验证集对训练好的模型进行验证；（A6）使用测试集对最终确定的模型进行测试。本发明具有识别准确等优点。