Deep Learning-Based Water Segmentation for Autonomous Surface Vessel

Abstract

Visual-based obstacle detection from an autonomous surface vessel (ASV) is a complex task due to high variance of scene properties such as different illumination and presence of reflections. One approach in implementing the task is through extracting waterlines to enable inferring of vessel orientation and obstacles presence. Classical computer vision algorithms for detection holds limitation in robustness and scalability. With recent breakthroughs in deep neural network architectures, vision-based object detection is seen to obtain high performance. In this work, the deep learning models based on Convolutional Neural Network (CNN) to implement binary semantic segmentation is studied. This architecture identifies each pixel to water and non-water classes. In purpose of benchmarking models, Fully Convolutional Network (FCN), SegNet and U-Net are trained on a publicly available dataset, IntCatch Vision Data Set (ICVDS), to evaluate the performance. From the experiments carried out, quantitative results show effectiveness of the models with accuracy all above 95.55% and lowest average speed of 11 frames per second. To improve, pre-trained networks (VGG 16, Resnet-50 and MobileNet) are used as a backbone, obtaining an improved accuracy above 98.14% with lowest inferring speed of 10 frame per second. Using the developed ASV, new dataset of 143 images called Malaysia ASV Dataset (MASVD) is collected, labelled and made publicly available. The trained models are tested with the newly collected dataset obtaining accuracy of 75%. The high accuracy performance shows potential for the models to be employed for collision avoidance algorithm in ASV navigation.