BiliSakura commited on
Commit
ee5c69c
Β·
verified Β·
1 Parent(s): 2edca4c

Update all files for SegEarth-OV

Browse files
Files changed (1) hide show
  1. README.md +115 -0
README.md ADDED
@@ -0,0 +1,115 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: mit
3
+ pipeline_tag: image-segmentation
4
+ tags:
5
+ - remote-sensing
6
+ - earth-observation
7
+ - open-vocabulary
8
+ - clip
9
+ - sam3
10
+ - semantic-segmentation
11
+ library_name: transformers
12
+ ---
13
+
14
+ # SegEarth-OV: Unified Open-Vocabulary Segmentation for Remote Sensing
15
+
16
+ Unified repo for **SegEarth OV, OV-2, OV-3** β€” training-free open-vocabulary semantic segmentation. Each variant lives in a **self-contained subfolder** with its own `config.json`, pipeline, and weights.
17
+
18
+ ## Structure
19
+
20
+ ```
21
+ SegEarth-OV/
22
+ β”œβ”€β”€ OV/ # CLIP (OpenAI ViT-B/16) + SimFeatUp
23
+ β”‚ β”œβ”€β”€ config.json
24
+ β”‚ β”œβ”€β”€ pipeline.py
25
+ β”‚ β”œβ”€β”€ upsamplers.py
26
+ β”‚ β”œβ”€β”€ prompts/
27
+ β”‚ β”œβ”€β”€ configs/cls_*.txt
28
+ β”‚ β”œβ”€β”€ weights/featup/ # SimFeatUp checkpoints
29
+ β”‚ └── weights/backbone/ # OpenAI CLIP (clip-vit-base-patch16)
30
+ β”œβ”€β”€ OV-2/ # AlignEarth (SAR) + SimFeatUp
31
+ β”‚ β”œβ”€β”€ config.json
32
+ β”‚ β”œβ”€β”€ pipeline.py
33
+ β”‚ β”œβ”€β”€ weights/featup/
34
+ β”‚ └── weights/backbone/ # AlignEarth-SAR-ViT-B-16
35
+ β”œβ”€β”€ OV-3/ # SAM3 (no featup)
36
+ β”‚ β”œβ”€β”€ config.json
37
+ β”‚ β”œβ”€β”€ pipeline.py
38
+ β”‚ β”œβ”€β”€ configs/
39
+ β”‚ └── weights/backbone/ # facebook/sam3 (sam3.pt)
40
+ └── model_config.json
41
+ ```
42
+
43
+ ## Self-Contained (No Download)
44
+
45
+ All checkpoints are **included** in this repo. No additional download required.
46
+
47
+ | Variant | Backbone | Location |
48
+ |---------|----------|----------|
49
+ | OV | OpenAI CLIP ViT-B/16 | `OV/weights/backbone/clip-vit-base-patch16/` |
50
+ | OV-2 | AlignEarth-SAR-ViT-B-16 | `OV-2/weights/backbone/AlignEarth-SAR-ViT-B-16/` |
51
+ | OV-3 | SAM3 | `OV-3/weights/backbone/sam3/sam3.pt` |
52
+ | OV, OV-2 | SimFeatUp (jbu_one, etc.) | `OV/weights/featup/`, `OV-2/weights/featup/` |
53
+
54
+ ## Usage
55
+
56
+ **From subfolder (self-contained):**
57
+
58
+ ```python
59
+ # OV-2 with AlignEarth (SAR)
60
+ from pipeline import SegEarthPipeline
61
+ pipe = SegEarthPipeline() # loads OV-2/config.json
62
+ seg = pipe(image)
63
+
64
+ # Or: cd OV-2 && python -c "from pipeline import load; pipe = load()"
65
+ ```
66
+
67
+ **From repo root:**
68
+
69
+ ```python
70
+ from pipeline import SegEarthPipeline
71
+
72
+ pipe = SegEarthPipeline(variant="OV-2")
73
+ pipe = SegEarthPipeline(variant="OV")
74
+ pipe = SegEarthPipeline(variant="OV-3") # requires sam3 package
75
+ ```
76
+
77
+ **Custom class config:**
78
+
79
+ ```python
80
+ pipe = SegEarthPipeline(variant="OV-2", class_names_path="OV-2/configs/cls_yeseg_sar.txt")
81
+ ```
82
+
83
+ ## Variants
84
+
85
+ | Subfolder | Backbone | Model ID | FeatUp |
86
+ |-----------|----------|----------|--------|
87
+ | OV | CLIP | openai/clip-vit-base-patch16 | jbu_one |
88
+ | OV-2 | AlignEarth | BiliSakura/AlignEarth-SAR-ViT-B-16 | jbu_one |
89
+ | OV-3 | SAM3 | facebook/sam3 | None |
90
+
91
+ ## Citation
92
+
93
+ ```bibtex
94
+ @InProceedings{Li_2025_CVPR,
95
+ author = {Li, Kaiyu and Liu, Ruixun and Cao, Xiangyong and Bai, Xueru and Zhou, Feng and Meng, Deyu and Wang, Zhi},
96
+ title = {SegEarth-OV: Towards Training-Free Open-Vocabulary Segmentation for Remote Sensing Images},
97
+ booktitle = {Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR)},
98
+ year = {2025},
99
+ pages = {10545--10556}
100
+ }
101
+
102
+ @article{li2025segearthov2,
103
+ title = {Annotation-Free Open-Vocabulary Segmentation for Remote-Sensing Images},
104
+ author = {Li, Kaiyu and Cao, Xiangyong and Liu, Ruixun and Wang, Shihong and Jiang, Zixuan and Wang, Zhi and Meng, Deyu},
105
+ journal = {arXiv preprint arXiv:2508.18067},
106
+ year = {2025}
107
+ }
108
+
109
+ @article{li2025segearthov3,
110
+ title = {SegEarth-OV3: Exploring SAM 3 for Open-Vocabulary Semantic Segmentation in Remote Sensing Images},
111
+ author = {Li, Kaiyu and Zhang, Shengqi and Deng, Yupeng and Wang, Zhi and Meng, Deyu and Cao, Xiangyong},
112
+ journal = {arXiv preprint arXiv:2512.08730},
113
+ year = {2025}
114
+ }
115
+ ```