地理空间信息2024,Vol.22Issue(4):45-48,4.DOI:10.3969/j.issn.1672-4623.2024.04.011
利用铁塔视频图像和改进YOLOv5的违规施工监测
Illegal Construction Monitoring Using Tower-based Video Image and Improved YOLOv5
余添添 1吴松 1唐芝青 2屈伟军 2肖德勋 1李琪1
作者信息
- 1. 北京吉威数源信息技术有限公司,北京 100000
- 2. 湖南省第二测绘院,湖南 长沙 410000
- 折叠
摘要
Abstract
The construction equipment targets such as excavators in the tower-based video are extremely sparse,mostly small and with complex backgrounds,and part of them is occluded.So,we proposed an improved detection algorithm based on YOLOv5s.Firstly,we changed the data augmentation mode to Mosain-9,to increase the number of small target.Then,we optimized the process of feature extraction and feature fusion,and added a detection scale to improve the detection ability of small targets and feature fusion.Finally,we introduced an attention mechanism to pay more attention to the region of interest,suppress useless information and background noise,and improve the accuracy of target detection.The experimental results show that AP value of the improved algorithm reaches 79.63%,which is 6.8%higher than that of YOLOv5s.At the same time,the detection accuracy of small targets is higher,and the missing target is less.关键词
工程机械检测/YOLOv5s/小目标检测/注意力机制Key words
construction equipment detection/YOLOv5s/small target/attention mechanism分类
天文与地球科学引用本文复制引用
余添添,吴松,唐芝青,屈伟军,肖德勋,李琪..利用铁塔视频图像和改进YOLOv5的违规施工监测[J].地理空间信息,2024,22(4):45-48,4.