Update README.md
Browse files
README.md
CHANGED
|
@@ -17,17 +17,9 @@ This model is a modified version of [RCAN](https://arxiv.org/abs/1807.02758), or
|
|
| 17 |
## 🧠 Model Description
|
| 18 |
|
| 19 |
- Based on the original RCAN architecture from `super-image`.
|
| 20 |
-
- `sub_mean` and `add_mean` normalization layers have been **removed
|
| 21 |
- Supports **multi-channel inputs**, currently set up for **2-channel wind fields**.
|
| 22 |
-
- Custom head and tail layers allow flexible channel dimensions and upscaling factors (e.g., ×2, ×4).
|
| 23 |
-
- Optional partial weight loading from pretrained RCAN (trained on DIV2K) to initialize the main body.
|
| 24 |
|
| 25 |
-
## 🛰️ Intended Use
|
| 26 |
-
|
| 27 |
-
This model is intended for:
|
| 28 |
-
- **Super-resolution of wind fields** in climate modeling applications.
|
| 29 |
-
- **Downscaling wind speed or wind power-related variables** from coarse-resolution reanalysis or GCM output.
|
| 30 |
-
- Research on generalizing deep learning architectures to geoscience and Earth system science domains.
|
| 31 |
|
| 32 |
## 🧪 Example
|
| 33 |
|
|
@@ -38,11 +30,8 @@ import torch
|
|
| 38 |
|
| 39 |
path = hf_hub_download(repo_id="lschmidt/rcan-dsc", filename="rcan_model.py")
|
| 40 |
|
| 41 |
-
# Dynamically load it
|
| 42 |
exec(open(path).read())
|
| 43 |
|
| 44 |
-
|
| 45 |
-
# Now you can call the function directly
|
| 46 |
model = load_rcan()
|
| 47 |
|
| 48 |
# load config
|
|
@@ -53,6 +42,5 @@ state_dict_path = hf_hub_download(repo_id="lschmidt/rcan-dsc", filename="pytorch
|
|
| 53 |
state_dict = torch.load(state_dict_path, map_location="cpu")
|
| 54 |
model.load_state_dict(state_dict, strict=False)
|
| 55 |
|
| 56 |
-
inputs = torch.randn(1, 2, 10, 10)
|
| 57 |
output = model(inputs)
|
| 58 |
-
|
|
|
|
| 17 |
## 🧠 Model Description
|
| 18 |
|
| 19 |
- Based on the original RCAN architecture from `super-image`.
|
| 20 |
+
- `sub_mean` and `add_mean` normalization layers have been **removed**
|
| 21 |
- Supports **multi-channel inputs**, currently set up for **2-channel wind fields**.
|
|
|
|
|
|
|
| 22 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 23 |
|
| 24 |
## 🧪 Example
|
| 25 |
|
|
|
|
| 30 |
|
| 31 |
path = hf_hub_download(repo_id="lschmidt/rcan-dsc", filename="rcan_model.py")
|
| 32 |
|
|
|
|
| 33 |
exec(open(path).read())
|
| 34 |
|
|
|
|
|
|
|
| 35 |
model = load_rcan()
|
| 36 |
|
| 37 |
# load config
|
|
|
|
| 42 |
state_dict = torch.load(state_dict_path, map_location="cpu")
|
| 43 |
model.load_state_dict(state_dict, strict=False)
|
| 44 |
|
| 45 |
+
inputs = torch.randn(1, 2, 10, 10)
|
| 46 |
output = model(inputs)
|
|
|