WaSR——一种用于海上障碍物检测的水体分割与细化网络

WaSR—A Water Segmentation and Refinement Maritime Obstacle Detection Network

IEEE Transactions on Cybernetics · 2021
被引 100
ABS 3

中文导读

针对水面反射和尾迹导致误检的问题,提出一种融合惯性测量单元信息的深度编码器-解码器网络WaSR,在无人水面艇数据集上F1分数提升约4%,并在域泛化实验中提升超24%。

Abstract

Obstacle detection using semantic segmentation has become an established approach in autonomous vehicles. However, existing segmentation methods, primarily developed for ground vehicles, are inadequate in an aquatic environment as they produce many false positive (FP) detections in the presence of water reflections and wakes. We propose a novel deep encoder-decoder architecture, a water segmentation and refinement (WaSR) network, specifically designed for the marine environment to address these issues. A deep encoder based on ResNet101 with atrous convolutions enables the extraction of rich visual features, while a novel decoder gradually fuses them with inertial information from the inertial measurement unit (IMU). The inertial information greatly improves the segmentation accuracy of the water component in the presence of visual ambiguities, such as fog on the horizon. Furthermore, a novel loss function for semantic separation is proposed to enforce the separation of different semantic components to increase the robustness of the segmentation. We investigate different loss variants and observe a significant reduction in FPs and an increase in true positives (TPs). Experimental results show that WaSR outperforms the current state of the art by approximately 4% in F1 score on a challenging unmanned surface vehicle dataset. WaSR shows remarkable generalization capabilities and outperforms the state of the art by over 24% in F1 score on a strict domain generalization experiment.

语义分割深度学习计算机视觉自动驾驶海洋环境