水力发电学报2025,Vol.44Issue(3):87-98,12.DOI:10.11660/slfdxb.20250308
深度学习驱动的无人机河道巡检异物监测及定位技术研究
Research on foreign object detection and localization in UAV river patrol driven by deep learning
摘要
Abstract
As the city expands,the river shoreline is constantly being encroached upon,and the river's flow capacity and water environment are damaged severely.Therefore,efficient methods for monitoring complicated river courses are urgently needed.This paper precents a new method for collecting images and data of a river and its banks using drones and enhancing data with the Generative Adversarial Network.We identify and locate five kinds of typical foreign bodies on the river,based on the YOLOv5 algorithm and the coordinate transformation localization algorithm.The target recognition algorithm of this model introduces the attention mechanism into the backbone network and uses EIOU-Focal Loss as its loss function to improve YOLOv5 in detection accuracy and convergence speed.The results show that data enhancement improves the model's target recognition and raises the mean Average Precision(mAP)by 9.9%.The ablation experiment results verify that this model has the highest detection accuracy,with its maximum mAP of 0.96 or an increase of 11.6%relative to the one before improvement.Its positioning results show the real average error of the algorithm target object is not greater than 3m,which means a high accuracy.Application of the improved model to the Minjiang River section in Fujian has verified its higher accuracy in detecting target objects and its significance for related research.关键词
河道检测/异物识别/深度学习/目标检测算法/生成对抗网络Key words
river detection/foreign body recognition/deep learning/YOLOv5/DCGAN分类
计算机与自动化引用本文复制引用
郑荣添,陈泽同,管孝汉,王浩,刘荣林,假冬冬..深度学习驱动的无人机河道巡检异物监测及定位技术研究[J].水力发电学报,2025,44(3):87-98,12.基金项目
福建省交通运输科技项目(ZH202309) (ZH202309)
福建省自然科学基金(2023J01406) (2023J01406)