Inshore Warship Detection Method Based on Multi-task Learning
In the task of inshore warship detection in remote sensing optical images,this paper proposes an inshore warship de-tection method based on multi-task learning for the false alarms problem of similar features in complex scenes.By constructing a parallel dual-branch task framework for the sea-land segmentation mission and the warship detection mission,this method opti-mizes the traditional task of serial processing into parallel processing mode.Secondly,we propose a joint loss constraint for dual path optimum training,which improves the stability of model training.Finally,the dataset made by Google Earth remote sensing images is used for experiments.The detection results in land mask are eliminated by the dual-branch fusion model,and the land false alarm filter is realized.Compared with the single task detection algorithm YOLOv5,the mAP of the proposed method in-creased by 4.4 percentage points and the false alarm rate decreased by 3.4 percentage points.The experimental results show that the proposed algorithm is effective in suppressing false alarm on land.
warship detectionsea-land segmentationmulti-task learningloss function
国家科技期刊平台
NSTL
万方数据
