Add model card, pipeline tag and library metadata

#1
by nielsr HF Staff - opened
Files changed (1) hide show
  1. README.md +41 -3
README.md CHANGED
@@ -1,3 +1,41 @@
1
- ---
2
- license: apache-2.0
3
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ pipeline_tag: image-segmentation
4
+ library_name: transformers
5
+ ---
6
+
7
+ # UniGeoSeg: Towards Unified Open-World Segmentation for Geospatial Scenes
8
+
9
+ UniGeoSeg is a unified framework for open-world segmentation in geospatial scenes. It is designed to handle various instruction-driven segmentation tasks in remote sensing, including referring, interactive, and reasoning-based segmentation.
10
+
11
+ - **Paper:** [UniGeoSeg: Towards Unified Open-World Segmentation for Geospatial Scenes](https://huggingface.co/papers/2511.23332)
12
+ - **Repository:** [GitHub - MiliLab/UniGeoSeg](https://github.com/MiliLab/UniGeoSeg)
13
+ - **Benchmark:** [GeoSeg-Bench](https://huggingface.co/datasets/nishuo1999/GeoSeg-Bench)
14
+
15
+ ## Introduction
16
+ UniGeoSeg addresses the challenges of fragmented task formulations and limited instruction data in remote sensing by leveraging a progressive training strategy and a unified architecture. It is trained on the **GeoSeg-1M** dataset, which contains 1.1 million image-mask-instruction triplets, providing strong zero-shot generalization capabilities across complex geospatial scenes.
17
+
18
+ The model incorporates task-aware text enhancement and latent knowledge memory to facilitate multi-task learning, achieving state-of-the-art performance across diverse geospatial benchmarks.
19
+
20
+ ## Usage
21
+
22
+ For inference and evaluation, please refer to the scripts provided in the official repository. You can run the provided inference script using:
23
+
24
+ ```bash
25
+ python scripts/eval.sh
26
+ ```
27
+
28
+ ## Citation
29
+ If you find UniGeoSeg helpful, please cite the following work:
30
+
31
+ ```bibtex
32
+ @misc{ni2025unigeosegunifiedopenworldsegmentation,
33
+ title={UniGeoSeg: Towards Unified Open-World Segmentation for Geospatial Scenes},
34
+ author={Shuo Ni and Di Wang and He Chen and Haonan Guo and Ning Zhang and Jing Zhang},
35
+ year={2025},
36
+ eprint={2511.23332},
37
+ archivePrefix={arXiv},
38
+ primaryClass={cs.CV},
39
+ url={https://arxiv.org/abs/2511.23332},
40
+ }
41
+ ```