Detection Method of Marine Biological Objects Based on Image Enhancement and Improved YOLOv5S

Abstract

Marine biological object detection is of great significance for the exploration and protection of underwater resources. There have been some achievements in visual inspection for specific objects based on machine learning. However, owing to the complex imaging environment, some problems, such as low accuracy and poor real-time performance, have appeared in these object detection methods. To solve these problems, this paper proposes a detection method of marine biological objects based on image enhancement and YOLOv5S. Contrast-limited adaptive histogram equalization is taken to solve the problems of underwater image distortion and blur, and we put forward an improved YOLOv5S to improve accuracy and real-time performance of object detection. Compared with YOLOv5S, coordinate attention and adaptive spatial feature fusion are added in the improved YOLOv5S, which can accurately locate the target of interest and fully fuse the features of different scales. In addition, soft non-maximum suppression is adopted to replace non-maximum suppression for the improvement of the detection ability for overlapping objects. The experimental results show that the contrast-limited adaptive histogram equalization algorithm can effectively improve the underwater image quality and the detection accuracy. Compared with the original model (YOLOv5S), the proposed algorithm has a higher detection accuracy. The detection accuracy AP50 reaches 94.9% and the detection speed is 82 frames per second; therefore, the real-time performance can be said to reach a high level.