zhangrenchao commited on
Commit
d534ae8
·
verified ·
1 Parent(s): 2d7eead

Update English model card

Browse files
Files changed (1) hide show
  1. README.md +47 -52
README.md CHANGED
@@ -1,19 +1,16 @@
1
  ---
2
- frameworks: PyTorch
3
  language:
4
  - en
5
- license: apache-2.0
6
  tags:
7
  - OneScience
8
  - Earth Science
9
- - Remote Sensing
10
  - Multi-Modal Remote Sensing
11
  - Semantic Segmentation
12
- - SkySense
13
- - arxiv:2312.10115
14
- tasks: []
15
  datasets: []
16
  ---
 
17
  <p align="center">
18
  <strong>
19
  <span style="font-size: 30px;">SkySense</span>
@@ -22,50 +19,48 @@ datasets: []
22
 
23
  # Model Introduction
24
 
25
- SkySense is a multi-modal and temporal remote sensing foundation model for universal Earth observation interpretation. It jointly models high-resolution optical imagery, Sentinel-2 multispectral time series, Sentinel-1 SAR time series, and geographic context to provide unified representations for remote sensing classification, segmentation, detection, and change detection.
26
-
27
- Paper: SkySense: A Multi-Modal Remote Sensing Foundation Model Towards Universal Interpretation for Earth Observation Imagery
28
 
 
29
  https://arxiv.org/abs/2312.10115
30
 
31
  # Model Description
32
 
33
- SkySense was proposed by Ant Group, Wuhan University, and MYbank. The original model was trained on 21.5 million groups of high-resolution optical, Sentinel-2 multispectral, and Sentinel-1 SAR temporal samples, and is suitable for single-modal or multi-modal, static or temporal remote sensing classification and localization tasks.
34
 
35
  # Use Cases
36
 
37
  | Scenario | Description |
38
  | :---: | :--- |
39
- | Multi-modal remote sensing representation training | Train SkySense with multi-modal temporal remote sensing image data. |
40
- | Remote sensing image semantic segmentation | Fuse optical, multispectral, and SAR features to predict land-cover classes for image pixels. |
41
- | Local quick validation | Use synthetic data to validate data loading, training, inference, evaluation, and visualization. |
42
- | Hugging Face / OneCode execution | Download the standalone model package, install dependencies, and run the scripts directly. |
43
- | Multi-GPU training | Launch distributed multi-process training with `torchrun`. |
 
44
 
45
  # Usage Guide
46
 
47
- ## 1. OneCode Usage
48
 
49
  Experience intelligent one-click AI4S programming through the OneCode online environment:
50
 
51
  [Click to Experience Intelligent One-Click AI4S Programming](https://web-2069360198568017922-iaaj.ksai.scnet.cn:58043/home)
52
 
53
- ## 2. Manual Installation and Usage
54
-
55
- **Hardware Requirements**
56
-
57
- - A GPU or DCU is recommended.
58
- - CPU can be used for import and small-scale connectivity verification; full training and inference will be slower.
59
- - DCU users must install DTK in advance. DTK 25.04.2 or above, or the OneScience recommended version matching the cluster, is recommended.
60
-
61
- ### Download the Model Package
62
 
63
  ```bash
64
  hf download OneScience-Group/SkySense --local-dir ./SkySense
65
  cd SkySense
66
  ```
67
 
68
- ### Install the Runtime Environment
 
 
 
 
 
 
69
 
70
  **DCU Environment**
71
 
@@ -89,42 +84,45 @@ pip install onescience[earth-gpu] -i http://mirrors.onescience.ai:3141/pypi/simp
89
 
90
  ### Training Data Introduction
91
 
92
- The original SkySense pretraining collection contains about 21.5 million spatially registered multi-modal remote sensing samples. It includes high-resolution optical imagery, Sentinel-2 multispectral temporal imagery, Sentinel-1 SAR temporal imagery, acquisition dates, geographic region information, and downstream task labels. The complete registered pretraining collection is not publicly available from the authors, and it is not included in this model repository.
93
 
94
- Original data sources include Sentinel-1 and Sentinel-2 products available through Copernicus Data Space and commercial high-resolution imagery such as WorldView. Users must follow the applicable data licenses and independently complete cloud filtering, radiometric processing, spatial registration, resampling, cropping, normalization, and label preparation.
95
 
96
- The default local validation uses synthetic NPZ data:
97
 
98
- ```bash
99
- python scripts/fake_data.py
 
 
 
 
 
 
 
 
 
 
100
  ```
101
 
102
- The command creates 24 training samples and 8 test samples. Synthetic data is used only to validate the complete training, inference, evaluation, and visualization pipeline. Synthetic results are not reported as model performance and do not reproduce the paper experiments.
103
-
104
- For real-data training, prepare the corresponding multi-modal samples in the format expected by the scripts, save them under `data/`, update the data and model settings in `conf/config.yaml`, and use the same `scripts/train.py`. Do not run `scripts/fake_data.py` when using real data.
105
-
106
- ### Training
107
-
108
- Synthetic-data quick validation:
109
 
110
  ```bash
111
  python scripts/fake_data.py
112
- python scripts/train.py
113
  ```
114
 
115
- Single GPU or CPU:
116
 
117
  ```bash
118
  python scripts/train.py
119
  ```
120
 
121
- Multi-GPU:
122
 
123
  ```bash
124
- torchrun --nproc_per_node=8 --nnodes=1 --rdzv_id=1000 --rdzv_backend=c10d --max_restarts=0 --master_addr="localhost" --master_port=29500 scripts/train.py
125
  ```
126
 
127
- Training outputs:
128
 
129
  ```text
130
  result/checkpoints/skysense.pt
@@ -133,7 +131,7 @@ result/training/metrics.json
133
 
134
  ### Training Weights
135
 
136
- Pretrained weights based on multi-modal temporal remote sensing data will be provided under `weight/`. The weight file will be uploaded in a future update. Checkpoints generated from synthetic data are only for pipeline validation and do not provide real remote sensing interpretation capability.
137
 
138
  ### Inference
139
 
@@ -141,9 +139,7 @@ Pretrained weights based on multi-modal temporal remote sensing data will be pro
141
  python scripts/inference.py
142
  ```
143
 
144
- Inference reads `data/test.npz` and `result/checkpoints/skysense.pt`. Test data must use the same channel order, normalization, spatial size, region mapping, and class definition as the training data.
145
-
146
- Prediction output:
147
 
148
  ```text
149
  result/output/
@@ -155,15 +151,13 @@ result/output/
155
  python scripts/result.py
156
  ```
157
 
158
- Evaluation and visualization outputs:
159
 
160
  ```text
161
  result/evaluation/metrics.json
162
  result/evaluation/comparison.png
163
  ```
164
 
165
- For synthetic data, evaluation outputs only confirm that the evaluation and visualization pipeline can run successfully. They are not reported as model performance. For real data, the metrics represent supervised semantic segmentation performance on the user-provided test set and are not equivalent to the complete 21.5-million-sample pretraining evaluation in the paper.
166
-
167
  # Official OneScience Resources
168
 
169
  | Platform | OneScience Main Repository | Skills Repository |
@@ -173,5 +167,6 @@ For synthetic data, evaluation outputs only confirm that the evaluation and visu
173
 
174
  # Citation and License
175
 
176
- - This repository is a reproduction of the original SkySense paper.
177
- - The reproduction code is released under Apache License 2.0. The original datasets and official model weights remain subject to their respective source licenses.
 
 
1
  ---
2
+ license: other
3
  language:
4
  - en
 
5
  tags:
6
  - OneScience
7
  - Earth Science
 
8
  - Multi-Modal Remote Sensing
9
  - Semantic Segmentation
10
+ frameworks: PyTorch
 
 
11
  datasets: []
12
  ---
13
+
14
  <p align="center">
15
  <strong>
16
  <span style="font-size: 30px;">SkySense</span>
 
19
 
20
  # Model Introduction
21
 
22
+ SkySense is a multimodal remote sensing foundation model for general Earth observation interpretation. It jointly represents high-resolution RGB, Sentinel-1, and Sentinel-2 data through independent spatial encoding, temporal aggregation, cross-modal fusion, and geographic prototype modeling.
 
 
23
 
24
+ Paper: SkySense: A Multi-Modal Remote Sensing Foundation Model Towards Universal Interpretation for Earth Observation Imagery
25
  https://arxiv.org/abs/2312.10115
26
 
27
  # Model Description
28
 
29
+ SkySense was proposed by a research team from Wuhan University, Huazhong University of Science and Technology, and other institutions. The model is trained with large-scale multimodal remote sensing time-series data and fuses high-resolution RGB, Sentinel-1 radar, and Sentinel-2 multispectral observations. It is suitable for multimodal remote sensing representation learning, land-cover semantic segmentation, and other Earth observation interpretation tasks.
30
 
31
  # Use Cases
32
 
33
  | Scenario | Description |
34
  | :---: | :--- |
35
+ | Multimodal remote sensing fusion | Fuse multi-resolution observations from HR, Sentinel-1, and Sentinel-2. |
36
+ | Multi-temporal modeling | Process satellite sequences using date encoding and intra-modality temporal aggregation. |
37
+ | Semantic segmentation | Produce high-resolution land-cover class maps. |
38
+ | Geographic object detection | Transfer and fine-tune multimodal representations for detecting remote sensing objects such as aircraft, ships, and vehicles. |
39
+ | Land-surface change detection | Adapt multi-temporal observations to downstream tasks to identify changes in buildings and land-cover regions. |
40
+ | Local engineering validation | Use a small amount of synthetic data to check the training, inference, and evaluation workflows. |
41
 
42
  # Usage Guide
43
 
44
+ ## 1. OneCode
45
 
46
  Experience intelligent one-click AI4S programming through the OneCode online environment:
47
 
48
  [Click to Experience Intelligent One-Click AI4S Programming](https://web-2069360198568017922-iaaj.ksai.scnet.cn:58043/home)
49
 
50
+ ## 2. Download and Installation
 
 
 
 
 
 
 
 
51
 
52
  ```bash
53
  hf download OneScience-Group/SkySense --local-dir ./SkySense
54
  cd SkySense
55
  ```
56
 
57
+ ### Environment Dependencies
58
+
59
+ **Hardware Requirements**
60
+
61
+ - A GPU or DCU is recommended.
62
+ - CPU can be used for small-configuration connectivity validation; full training and inference will be slow.
63
+ - DCU users must install DTK in advance. DTK 25.04.2 or above, or the OneScience recommended version matching the current cluster, is recommended.
64
 
65
  **DCU Environment**
66
 
 
84
 
85
  ### Training Data Introduction
86
 
87
+ By default, 2 training and 1 test synthetic samples are used to validate the engineering workflow. They are saved as `data/train.npz` and `data/test.npz`, respectively.
88
 
89
+ The synthetic data preserves the specifications from the paper and the official backbone: static HR, 20 Sentinel-2 time steps, 10 Sentinel-1 time steps, and the channel counts and spatial sizes of each modality.
90
 
91
+ Real data must be preprocessed and converted to the following NPZ training protocol. This protocol is consistent with the model input specification but is not the download format of the original datasets.
92
 
93
+ ```text
94
+ hr: float32 [N,1,3,224,224]
95
+ s2: float32 [N,20,10,64,64]
96
+ s1: float32 [N,10,2,64,64]
97
+ dates_hr: int64 [N,1]
98
+ dates_s2: int64 [N,20]
99
+ dates_s1: int64 [N,10]
100
+ region: int64 [N]
101
+ labels: int64 [N,224,224]
102
+ band_order_hr: string [3]
103
+ band_order_s2: string [10]
104
+ band_order_s1: string [2]
105
  ```
106
 
107
+ `fake_data.py` automatically writes the `protocol` and `data_source` protocol metadata. These fields must be retained when using real data.
 
 
 
 
 
 
108
 
109
  ```bash
110
  python scripts/fake_data.py
 
111
  ```
112
 
113
+ ### Training
114
 
115
  ```bash
116
  python scripts/train.py
117
  ```
118
 
119
+ For multi-GPU training, use:
120
 
121
  ```bash
122
+ torchrun --nproc_per_node=8 scripts/train.py
123
  ```
124
 
125
+ Training jointly optimizes semantic segmentation and cross-modal representation alignment objectives and saves a checkpoint and aggregate training metrics. The default configuration is intended for quick workflow validation. Formal experiments should use the multimodal time-series data, model configuration, and training duration corresponding to the paper.
126
 
127
  ```text
128
  result/checkpoints/skysense.pt
 
131
 
132
  ### Training Weights
133
 
134
+ This repository will provide SkySense training weights in the `weight/` folder. The weight files will be uploaded soon and are expected to be available in the near future.
135
 
136
  ### Inference
137
 
 
139
  python scripts/inference.py
140
  ```
141
 
142
+ Inference loads the training checkpoint, generates high-resolution semantic segmentation results in batches, and saves them to:
 
 
143
 
144
  ```text
145
  result/output/
 
151
  python scripts/result.py
152
  ```
153
 
154
+ Evaluation reports pixel accuracy, per-class IoU, and mean IoU, and generates a comparison figure of input, label, and prediction. Results on synthetic data are only for engineering workflow validation and do not represent full-paper performance.
155
 
156
  ```text
157
  result/evaluation/metrics.json
158
  result/evaluation/comparison.png
159
  ```
160
 
 
 
161
  # Official OneScience Resources
162
 
163
  | Platform | OneScience Main Repository | Skills Repository |
 
167
 
168
  # Citation and License
169
 
170
+ This repository is a reproduction of the original SkySense paper.
171
+
172
+ Use of the code and data in this repository remains subject to the licenses and terms of use of their respective projects.