Differentiation of River Sediments Fractions in UAV Aerial Images by Convolution Neural Network

Surface sediment classification using a deep learning model and unmanned aerial vehicle data of tidal flats

This study proposes a deep learning model, U-Net, to improve surface sediment classification using high-resolution unmanned aerial vehicle (UAV) images. We constructed training datasets with UAV images and corresponding labeling data acquired from three field surveys on the Hwangdo tidal flat. The labeling data indicated the distribution of surface sediment types. We compared the performance of the U-Net model trained in various implementation environments, such as surface sediment criteria, input datasets, and classification models. The U-Net trained with five class criteria-derived from previous classification criteria-yielded valid results (overall accuracy:65.6 %). The most accurate results were acquired from trained U-Net with all input datasets; in particular, the tidal channel density caused a significant increase in accuracy. The accuracy of the U-Net was approximately 20 % higher than that of other classification models. These results demonstrate that surface sediment classification using UAV images and the U-Net model is effective.