Underwater Image Saliency Detection Via Attention-based Mechanism

Abstract

Due to complex lighting circumstances and other factors, visual saliency recognition of underwater images is not as good as visual saliency detection models that function well on land when applied to underwater environments. The dearth of large-scale annotated underwater picture datasets has further hindered the development of visual saliency detection in underwater images. we propose an attention-based approach for underwater image saliency detection to address this issue. The backbone network is pre-trained using both land and underwater image datasets, which effectively solves the problem of fewer datasets. After that, a residual refinement module incorporating a channel-space attention mechanism is merged, the model as a whole is trained end-to-end using underwater image datasets, and the model accuracy is experimentally tested. The experimental results in three underwater image datasets show that our method is better than several other methods in the mainstream evaluation indicators of the saliency detection.