- Conference Article
1
- 10.1145/3584376.3584523
Research on underwater biological detection method based on multi-scale fusion and attention mechanism
- Dec 16, 2022
- Longyu Tang + 2 more +2
Underwater robots are an important tool for exploiting marine resources, while underwater target detection is a key technology for marine fishing by underwater robots. For the problem that the traditional algorithm is difficult to detect small targets and multiple scales in the underwater environment, this paper proposes an underwater biological detection algorithm based on the multi-scale feature fusion and attention mechanism. First, a YOLOv5 network is constructed, and a bi-directionally weighted BiFPN feature pyramid structure is fed into the neck network so that multiscale target features can be efficiently merged. Second, a channel and spatial convolutional attention mechanism is introduced between the neck network and the prediction network in order to increase the saliency of the underwater target that needs to be detected, which effectively reduces the missed and erroneous detection of underwater targets. The experimental results show that the proposed algorithm can accurately detect multi-scale underwater targets in complex underwater environments with an average accuracy of 80.18%, which is 2.87% better than YOLOv5, and an average recall of 85.06%, which is 4.81% better than YOLOv5.
Read more