Opto-Electronic Engineering, Volume. 51, Issue 4, 240025-1(2024)

PAW-YOLOv7: algorithm for detection of tiny floating objects in river channels

Qinglei Luan1...2, Xinyu Chang1,2, Ye Wu1,2, Conglong Deng1,2,*, Yanqiong Shi1,2, and Zihua Chen12 |Show fewer author(s)
Author Affiliations
  • 1School of Mechanical and Electrical Engineering, Anhui Jianzhu University, Hefei, Anhui 230601, China
  • 2Anhui Province Key Laboratory of Intelligent Manufacturing of Construction Machinery, Hefei, Anhui 230601, China
  • show less

    Detection of floating debris in rivers is of great significance for ship autopilot and river cleaning, but the existing methods in targeting floating objects in the river with small target sizes and mutual occlusion, and less feature information lead to low detection accuracy. To address these problems, this paper proposes a small target object detection method called PAW-YOLOv7 based on YOLOv7. Firstly, in order to improve the feature expression ability of the small target network model, a small target object detection layer is constructed, and the self-attention and convolution hybrid module (ACmix) is integrated and applied to the newly constructed small target detection layer. Secondly, in order to reduce the interference of the complex background, the Omni-dimensional dynamic convolution (ODConv) is used instead of the convolution module in the neck, so as to give the network the ability to capture the global contextual information. Finally, the PConv (partial convolution) module is integrated into the backbone network to replace part of the standard convolution, while the WIoU (Wise-IoU) loss function is used to replace the CIoU. It achieves the reduction of network model computation, improves the network detection speed, and increases the focusing ability on the low-quality anchor frames, accelerating the convergence speed of the model. The experimental results show that the detection accuracy of the PAW-YOLOv7 algorithm on the FloW-Img dataset improved by the data extension technique in this paper reaches 89.7%, which is 9.8% higher than that of the original YOLOv7, the detection speed reaches 54 frames per second (FPS), and the detection accuracy on the self-built sparse floater dataset improves by 3.7% compared with that of YOLOv7. It is capable of detecting the tiny floating objects in the river channel quickly and accurately, and also has a better real-time detection performance.

    Keywords
    Tools

    Get Citation

    Copy Citation Text

    Qinglei Luan, Xinyu Chang, Ye Wu, Conglong Deng, Yanqiong Shi, Zihua Chen. PAW-YOLOv7: algorithm for detection of tiny floating objects in river channels[J]. Opto-Electronic Engineering, 2024, 51(4): 240025-1

    Download Citation

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    Category: Article

    Received: Jan. 25, 2024

    Accepted: Mar. 12, 2024

    Published Online: Jul. 8, 2024

    The Author Email: Deng Conglong (邓从龙)

    DOI:10.12086/oee.2024.240025

    Topics