Two-stage ship detection at long distances based on deep learning and slicing technique.

Abstract

Ship detection over long distances is crucial for the visual perception of intelligent ships. Since traditional image processing-based methods are not robust, deep learning-based image recognition methods can automatically obtain the features of small ships. However, due to the limited pixels of ships over long distances, accurate features of such ships are difficult to obtain. To address this, a two-stage object detection method that combines the advantages of traditional and deep-learning methods is proposed. In the first stage, an object detection model for the sea-sky line (SSL) region is trained to select a potential region of ships. In the second stage, another object detection model for ships is trained using sliced patches containing ships. When testing, the SSL region is first detected using the trained 8th version of You Only Look Once (YOLOv8). Then, the SSL region detected is divided into several overlapping patches using the slicing technique, and another trained YOLOv8 is applied to detect ships. The experimental results showed that our method achieved 85% average precision when the intersection over union is 0.5 (AP50), and a detection speed of 75 ms per image with a pixel size of 1080×640. The code is available at https://github.com/gongyanfeng/PaperCode.