Home > Published Issues > 2024 > Volume 12, No. 1, 2024 >
JOIG 2024 Vol.12(1): 76-89
doi: 10.18178/joig.12.1.76-89

Classification of Water Bodies Using Ensemble of U-Net and Random Forest Algorithm

Vasavi S *, Venkata Kalyan Chintalapudi, and Akhila Sree Rajeswari Vuppuluri
Velagapudi Ramakrishna Siddhartha Engineering College, Andhra Pradesh, India
Email: vasavi.movva@gmail.com (V.S.); 208w1a0511@vrsec.ac.in (V.K.C.);
208w1a0560@vrsec.ac.in (A.S.R.V.)
*Corresponding author

Manuscript received August 22, 2023; revised October 8, 2023; accepted October 17, 2023; published March 8, 2024.

Abstract—Water bodies classification using remote sensing and deep learning techniques plays a pivotal role in the effective management of water resources. This study aims to address the challenge of accurate water body detection and classification, which is essential for understanding their distribution and characteristics, ultimately informing water usage and conservation efforts. Current methodologies predominantly rely on Support Vector Machines (SVM) and pixel-based approaches, resulting in suboptimal accuracy. In response, this paper proposes an ensemble model that combines the U-Net neural network and the Random Forest algorithm for enhanced water body detection and classification. The research commences by obtaining high-resolution satellite images with a resolution of 0.5 m. The U-Net model is employed to segment water bodies, and contour analysis is subsequently applied to extract shape features. The Random Forest Classifier is then utilized to classify the segmented water bodies into distinct categories, including rivers, ponds, lakes, canals, and other water bodies. Following the U-Net segmentation, the rasterized segments are converted into vector format. These vector data are leveraged to update Geographic Information System (GIS) maps, contributing to more accurate cartographic representations. The proposed approach is rigorously evaluated using a dataset from urban areas in Kolkata, West Bengal, India. The achieved accuracy rate stands at 67.01%.

Keywords—water body, shape features, U-Net, random forest algorithm, semantic segmentation, Geographic Information System (GIS) map

Cite: Vasavi S, Venkata Kalyan Chintalapudi, and Akhila Sree Rajeswari Vuppuluri, "Classification of Water Bodies Using Ensemble of U-Net and Random Forest Algorithm," Journal of Image and Graphics, Vol. 12, No. 1, pp. 76-89, 2024.

Copyright © 2024 by the authors. This is an open access article distributed under the Creative Commons Attribution License (CC BY-NC-ND 4.0), which permits use, distribution and reproduction in any medium, provided that the article is properly cited, the use is non-commercial and no modifications or adaptations are made.