Vessel Detection From Nighttime Remote Sensing Imagery Based on Deep Learning
Jiangnan Shao, Qingyao Yang, Changyu Luo, Ronghao Li, Yongsheng Zhou, Feixiang Zhang
Abstract
The continuous and rapid detection of sea vessels on a large scale is of great importance in marine traffic management, resource protection, and rights maintenance. Nighttime remote sensing can reflect human activities during the night with a wide swath and high efficiency, which is unique for vessel detection. Deep learning algorithms have already demonstrated superior performance in many fields, but it is confronted with some problems when applied to vessel detection with nighttime remote sensing imagery, including the lack of labeled dataset, the missed detection of small vessels, and false alarms of land targets. In this paper, firstly, the nighttime remote sensing imagery was collected and the sea vessels in it were labeled. Secondly, to enhance the detection performance of small vessels, a modified YOLOv5 algorithmTASFF-YOLOv5 was proposed, which was supplemented with a Tiny Target Detection Layer and a four-layer Adaptively Spatial Feature Fusion network to obtain better feature fusion. Thirdly, a land mask operation based on the sea-land prior database was performed to eliminate the false alarms of the land lights. Experiment results showed that the proposed TASFF-YOLOv5 could effectively improve the Precision, Recall, and mAP\@0.5 on the vessel dataset, achieving 95.2%, 93.1%, and 94.9% respectively.