一种基于改进YOLOv7的道路积水检测方法及装置

本发明公开一种基于改进YOLOv7的道路积水检测方法及装置，涉及城市安全算法技术领域。本发明提出了一种基于改进YOLOv7的道路积水检测方法,利用计算机视觉算法YOLOv7与Transformer结合,实现对积水图像和积水视频的实时检测。所述模型在原始YOLOv7模型的主干网络中嵌入Bottleneck Transformer结构，并对YOLOv7网络结构进行修改，增强模型在浅层网络对小尺寸积水点全局位置信息提取能力，提高小尺寸积水点深层网络中颜色、纹理等语义信息权重。