基于改进YOLO模型的堤防管涌险情智能识别方法

摘要

本发明公开了一种基于改进YOLO模型的堤防管涌险情智能识别方法，该方法包括改进和训练YOLO模型流程和基于改进YOLO模型堤防管涌识别的过程。改进和训练YOLO模型流程包括数据采集、制作样本、图像去背景处理和模型改进和训练步骤；基于改进YOLO模型识别的堤防管涌过程包括获取图像、划分窗口、图像处理与堤防管涌识别和输出最终识别结果等步骤。本发明利用单个YOLO模型实现堤防管涌目标特征提取、识别、目标定位等过程，克服人工特征提取的不足深层次挖掘数据所含信息；修改了原始YOLO模型，提升到小目标识别能力和速度，利用针对性的图像背景方法，分离大部分图像的背景，减少图像背景的干扰，提高堤防管涌识别精度。