The spillage on the expressway pavement have the characteristics of centralized distribution, which makes the small target scattered objects in the aerial images of unmanned aerial vehicles overlap, block and lack, which makes it difficult for the traditional target detection algorithm to capture its complex characteristics and can not adapt to the accurate detection of small target spillage. Based on the classic YOLOv8s detection model, an improved detection algorithm is proposed, which conforms to the distribution characteristics of small target spills on expressway pavement. The improved detection algorithm improves the detection accuracy of the algorithm by increasing the detection layer of small target spillage, introducing the cooperative attention mechanism of space and channel, and adjusting the loss function of dynamic non-monotonic focusing mechanism. The empirical analysis results show that the proposed improved detection algorithm has good detection performance, and the detection accuracy is improved by 9.9% compared with the traditional YOLO detection algorithm. Aiming at the sensitivity analysis of attention mechanism and loss function, the improved detection algorithm can improve the detection accuracy and recall rate and reduce the detection time.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

Small Target Spillage Detection Algorithm Based on Spatial Attention and Channel Attention Mechanism

  • Xindi Wu,
  • Jiangfeng Wang,
  • Shiqi Zhang,
  • Lu Ma

摘要

The spillage on the expressway pavement have the characteristics of centralized distribution, which makes the small target scattered objects in the aerial images of unmanned aerial vehicles overlap, block and lack, which makes it difficult for the traditional target detection algorithm to capture its complex characteristics and can not adapt to the accurate detection of small target spillage. Based on the classic YOLOv8s detection model, an improved detection algorithm is proposed, which conforms to the distribution characteristics of small target spills on expressway pavement. The improved detection algorithm improves the detection accuracy of the algorithm by increasing the detection layer of small target spillage, introducing the cooperative attention mechanism of space and channel, and adjusting the loss function of dynamic non-monotonic focusing mechanism. The empirical analysis results show that the proposed improved detection algorithm has good detection performance, and the detection accuracy is improved by 9.9% compared with the traditional YOLO detection algorithm. Aiming at the sensitivity analysis of attention mechanism and loss function, the improved detection algorithm can improve the detection accuracy and recall rate and reduce the detection time.