SO-YOLOv5: Small object recognition algorithm for sea cucumber in complex seabed environment

Abstract

Underwater object recognition is a prerequisite for the realization of automated seafood harvesting. To solve the problems of low recognition accuracy and poor generalization ability of existing underwater small object, this paper takes sea cucumber as research objects to research the small object recognition approach in complex underwater environment. In this paper, a sea cucumber dataset is established to solve the problem of lacking sea cucumber dataset in the real seabed environment. A deep learning model SO-YOLOv5 is proposed based on YOLOv5 for underwater small object recognition, which improves the recognition ability of the algorithm for different sizes of objects in complex environment. The proposed model embeds a large-scale feature extraction layer in structure to increase the detection ability of small objects by referring to the idea of the Bi-directional Feature Pyramid Network. SO-YOLOv5 fuses the features between deep and shallow layers and balances the feature information of different scales, and integrates the Coordinate Attention mechanism to enhance the sensitivity of the algorithm to the direction and position information. The experimental results illustrate that the mAP of the proposed approach achieves 95.47% for sea cucumber recognition in the complex seabed environment, which is 3.42%, 6.79%, and 5.46% higher than the traditional Faster R-CNN, SSD, and YOLOv5 algorithms, respectively. This research not only provides an effective approach for sea cucumber recognition in complex environment but also has certain reference significance for the recognition of small objects in other complex environments. In addition, the proposed approach has practical application value to improve intelligence level in aquaculture.