HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

U-Net Ensemble for Enhanced Semantic Segmentation in Remote Sensing Imagery

{Ivan Kitanovski Suzana Loshkovska Vlatko Spasev Ivica Dimitrovski}

Abstract

Semantic segmentation of remote sensing imagery stands as a fundamental task within the domains of both remote sensing and computer vision. Its objective is to generate a comprehensive pixel-wise segmentation map of an image, assigning a specific label to each pixel. This facilitates in-depth analysis and comprehension of the Earth’s surface. In this paper, we propose an approach for enhancing semantic segmentation performance by employing an ensemble of U-Net models with three different backbone networks: Multi-Axis Vision Transformer, ConvFormer, and EfficientNet. The final segmentation maps are generated through a geometric mean ensemble method, leveraging the diverse representations learned by each backbone network. The effectiveness of the base U-Net models and the proposed ensemble is evaluated on multiple datasets commonly used for semantic segmentation tasks in remote sensing imagery, including LandCover.ai, LoveDA, INRIA, UAVid, and ISPRS Potsdam datasets. Our experimental results demonstrate that the proposed approach achieves state-of-the-art performance, showcasing its effectiveness and robustness in accurately capturing the semantic information embedded within remote sensing images.

Benchmarks

BenchmarkMethodologyMetrics
semantic-segmentation-on-isprs-potsdamU-Net (ConvFormer-M36)
Mean IoU: 89.45
semantic-segmentation-on-landcover-aiU-Net (ConvFormer-M36)
mIoU: 87.64
semantic-segmentation-on-lovedaU-Net (MaxViT-S)
Category mIoU: 56.16
semantic-segmentation-on-uavidU-Net Ensemble
Mean IoU: 73.34
semantic-segmentation-on-uavidU-Net (MaxViT-S)
Mean IoU: 71.88

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp
U-Net Ensemble for Enhanced Semantic Segmentation in Remote Sensing Imagery | Papers | HyperAI