SegEarth-OV: Towards Training-Free Open-Vocabulary Segmentation for Remote Sensing Images
🌍SegEarth-OV: Towards Training-Free Open-Vocabulary Segmentation for Remote Sensing Images

— Make OVSS possible in remote sensing contexts

Kaiyu Li1
Ruixun Liu1
Xiangyong Cao✉ 1
Xueru Bai2
Feng Zhou2
Deyu Meng1
Zhi Wang1

Xi'an Jiaotong University1
Xidian University2

Code [GitHub]
Paper [arXiv]
Demo [Colab]


Visualization and performance of SegEarth-OV on open-vocabulary semantic segmentation of remote sensing images. We evaluate on 17 remote sensing datasets (including semantic segmentation, building extraction, road extraction, and flood detection tasks), and our SegEarth-OV consistently generates high-quality segmentation masks.



Abstract

Remote sensing image plays an irreplaceable role in fields such as agriculture, water resources, military, and disaster relief. Pixel-level interpretation is a critical aspect of remote sensing image applications; however, a prevalent limitation remains the need for extensive manual annotation. For this, we try to introduce open-vocabulary semantic segmentation (OVSS) into the remote sensing context. However, due to the sensitivity of remote sensing images to low-resolution features, distorted target shapes and ill-fitting boundaries are exhibited in the prediction mask. To tackle this issue, we propose a simple and general upsampler, SimFeatUp, to restore lost spatial information in deep features in a training-free style. Further, based on the observation of the abnormal response of local patch tokens to [CLS] token in CLIP, we propose to execute a straightforward subtraction operation to alleviate the global bias in patch tokens. Extensive experiments are conducted on 17 remote sensing datasets spanning semantic segmentation, building extraction, road detection, and flood detection tasks. Our method achieves an average of 5.8%, 8.2%, 4%, and 15.3% improvement over state-of-the-art methods on 4 tasks.



Pipeline


Illustration of the proposed method. (a) is the training process of SimFeatUp. CLIP is frozen and only SimFeatUp is useful in reasoning. (b) is the reasoning process of SegEarth-OV. The LR feature maps from CLIP are upsampled by SimFeatUp and then the [CLS] token is subtracted to alleviate global bias. For better presentation, the color renderings follow FeatUp.



Quantitative Results



Visualizations

Visualizations of the segmentation results

More Visualization ...

OpenEarthMap

UDD5

WHU Building



Acknowledgements

Based on a template by Phillip Isola and Richard Zhang.