zhangrenchao commited on
Commit
8c20e29
·
verified ·
1 Parent(s): 63ef347

Add English model card

Browse files
Files changed (1) hide show
  1. README.md +84 -0
README.md ADDED
@@ -0,0 +1,84 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ language:
4
+ - en
5
+ tags:
6
+ - OneScience
7
+ - Earth Science
8
+ - Extreme Weather
9
+ - Semantic Segmentation
10
+ - Tropical Cyclone
11
+ - Atmospheric River
12
+ frameworks: PyTorch
13
+ ---
14
+
15
+ <p align="center"><strong><span style="font-size: 30px;">ClimateNet</span></strong></p>
16
+
17
+ # Model Introduction
18
+
19
+ ClimateNet is an expert-labeled extreme-weather dataset and pixel-level segmentation model for tropical cyclones and atmospheric rivers.
20
+
21
+ Paper: ClimateNet: an expert-labeled open dataset and deep learning architecture for enabling high-precision analyses of extreme weather
22
+ https://doi.org/10.5194/gmd-14-107-2021
23
+
24
+ # Model Description
25
+
26
+ The model was proposed by teams from LBNL, UC Berkeley, ETH Zurich, NVIDIA, NCAR, and collaborators. It was trained with four-channel CAM5.1 fields and expert segmentation masks. DeepLabv3+ supports tropical-cyclone and atmospheric-river detection and conditional precipitation analysis.
27
+
28
+ # Use Cases
29
+
30
+ | Use Case | Description |
31
+ | :---: | :--- |
32
+ | Extreme segmentation | Identify background, TC, and AR pixels. |
33
+ | Climate scenarios | Transfer segmentation to warming experiments. |
34
+ | Conditional precipitation | Extract event-conditioned precipitation statistics. |
35
+ | ModelScope/OneCode execution | Validate data, training, inference, segmentation metrics, and visualization. |
36
+ | Multi-GPU training | Start multi-process training through `torchrun`. |
37
+
38
+ # Usage Instructions
39
+
40
+ ```bash
41
+ hf download OneScience-Group/ClimateNet --local-dir ./ClimateNet
42
+ cd ClimateNet
43
+ ```
44
+
45
+ ### Environment Dependencies
46
+
47
+ **Hardware Requirements**
48
+
49
+ - A GPU or DCU is recommended.
50
+ - A CPU can be used for connectivity validation with the default small-sample configuration.
51
+ - DCU users should install DTK 25.04.2 or a compatible OneScience-recommended version first.
52
+
53
+ **DCU Environment**
54
+
55
+ ```bash
56
+ # Activate DTK and Conda first
57
+ conda create -n onescience311 python=3.11 -y
58
+ conda activate onescience311
59
+ pip install onescience[earth-dcu] -i http://mirrors.onescience.ai:3141/pypi/simple/ --trusted-host mirrors.onescience.ai
60
+ ```
61
+
62
+ **GPU Environment**
63
+
64
+ ```bash
65
+ # Activate Conda first
66
+ conda create -n onescience311 python=3.11 -y libstdcxx-ng=12 libgcc-ng=12 gcc_linux-64=12 gxx_linux-64=12
67
+ conda activate onescience311
68
+ pip install onescience[earth-gpu] -i http://mirrors.onescience.ai:3141/pypi/simple/ --trusted-host mirrors.onescience.ai
69
+ ```
70
+
71
+ ```bash
72
+ python scripts/fake_data.py
73
+ python scripts/train.py
74
+ torchrun --standalone --nproc_per_node=2 scripts/train.py
75
+ python scripts/inference.py
76
+ python scripts/result.py
77
+ ```
78
+ Training uses weighted cross-entropy. Inference returns finite class probabilities and evaluation reports per-class and mean IoU.
79
+ ## Trained Weights
80
+ No weights are bundled under `weight/`. The authors provide trained models and data at https://portal.nersc.gov/project/ClimateNet/.
81
+ # Citation and License
82
+ This repository is an independent engineering reproduction of the public ClimateNet specifications.
83
+
84
+ The original paper is licensed under CC BY 4.0; official models, code, and data retain their respective terms.