For the difficult problem of small water floating object recognition, a deep learning-based target recognition method is improved. In this paper, an improved YOLOv5s target recognition algorithm is used to recognize water floating objects. First, according to the characteristics of the shape of floating objects on the water surface, the improved K-means algorithm is used to re-cluster the anchor boxes, then, Squeeze-and-Excitation networks module is added and then α-IOU is applied to the YOLOv5s network. The experimental results show that, compared with the standard YOLOv5s algorithm, the precision and average precision of the improved YOLOv5s algorithm are increased by 2% and 4% respectively, which verifies the effectiveness of the algorithm. This method can overcome the influence of water surface environment and effectively identify floating objects on the water surface.
刘麒, 尹港 , 王影, 叶泽 .
基于深度学习的水面漂浮物识别算法设计
[J]. 吉林化工学院学报, 2022, 39(7): 28-33.
LIU qi , YIN Gang , WANG ying , YE Ze.
Design of Water Surface Floating Object Recognition Algorithm based on Deep Learning
. Journal of Jilin Institute of Chemical Technology, 2022, 39(7): 28-33.