This repository contains the official implementation of the CVPR 2026 Findings paper:
"ConInfer: Context-Aware Inference for Training-Free Open-Vocabulary Remote Sensing Segmentation"
Training-free open-vocabulary remote sensing segmentation (OVRSS), empowered by vision-language models, has emerged as a promising paradigm for achieving category-agnostic semantic understanding in remote sensing imagery. Existing approaches mainly focus on enhancing feature representations or mitigating modality discrepancies to improve patch-level prediction accuracy. However, such independent prediction schemes are fundamentally misaligned with the intrinsic characteristics of remote sensing data. In real-world applications, remote sensing scenes are typically large-scale and exhibit strong spatial as well as semantic correlations, making isolated patch-wise predictions insufficient for accurate segmentation. To address this limitation, we propose \textbf{ConInfer}, a context-aware inference framework for OVRSS that performs joint prediction across multiple spatial units while explicitly modeling their inter-unit semantic dependencies. By incorporating global contextual cues, our method significantly enhances segmentation consistency, robustness, and generalization in complex remote sensing environments. Extensive experiments on multiple benchmark datasets demonstrate that our approach consistently surpasses state-of-the-art per-pixel VLM-based baselines such as SegEarth-OV, achieving average improvements of 2.80% and 6.13% on open-vocabulary semantic segmentation and object extraction tasks, respectively. All source codes will be publicly released upon the paper’s acceptance.
Clone the repository and install required packages.
git clone https://github.com/Dog-Yang/ConInfer.git
cd ConInfer
conda create -n ConInfer python=3.11
pip install -r requirements.txtWe include the following dataset configurations in this repo:
Semantic Segmentation: OpenEarthMap, LoveDA, iSAID, Potsdam, Vaihingen, UAVidimg, UDD5, VDDBuilding Extraction: WHUAerial, WHUSat.Ⅱ, Inria, xBDpreRoad Extraction: CHN6-CUG, DeepGlobe, Massachusetts, SpaceNetWater Extraction: WBS-SI
Please refer to dataset_prepare.md for dataset preparation.
Multi-GPU:
bash ./dist_test.sh ./config/cfg_DATASET.py
Results will be saved in results.xlsx.
- Thanks to @likyoo for their awesome SegEarth.
- Thanks to @facebookresearch for their awesome DINOV3.
