Upload README.md with huggingface_hub
Browse files
README.md
CHANGED
|
@@ -7,6 +7,7 @@ tags:
|
|
| 7 |
- eo-sar
|
| 8 |
- pytorch
|
| 9 |
- segmentation
|
|
|
|
| 10 |
datasets:
|
| 11 |
- doron333/change-detection-dataset
|
| 12 |
metrics:
|
|
@@ -14,6 +15,7 @@ metrics:
|
|
| 14 |
- precision
|
| 15 |
- recall
|
| 16 |
- f1
|
|
|
|
| 17 |
---
|
| 18 |
|
| 19 |
# Binary Change Detection (EO-SAR Fusion)
|
|
@@ -24,6 +26,7 @@ This model is a **Siamese UNet** designed for binary change detection using fuse
|
|
| 24 |
The architecture uses dual weight-shared **ResNet-34** encoders to extract multi-modal features from pre-event RGB (EO) and post-event grayscale (SAR) images. Feature differences are fused via skip connections into a UNet decoder.
|
| 25 |
|
| 26 |
- **Encoder:** ResNet-34 (Pre-trained on ImageNet)
|
|
|
|
| 27 |
- **Loss Function:** Combined Focal Loss + Dice Loss (Optimized for 58:1 class imbalance)
|
| 28 |
- **Input Resolution:** 256x256
|
| 29 |
|
|
@@ -38,3 +41,12 @@ This model is intended for rapid disaster damage assessment where pre-event opti
|
|
| 38 |
|
| 39 |
## How to Use
|
| 40 |
The model expects a 3-channel EO image (Pre-event) and a 1-channel SAR image (Post-event) as inputs.
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 7 |
- eo-sar
|
| 8 |
- pytorch
|
| 9 |
- segmentation
|
| 10 |
+
- safetensors
|
| 11 |
datasets:
|
| 12 |
- doron333/change-detection-dataset
|
| 13 |
metrics:
|
|
|
|
| 15 |
- precision
|
| 16 |
- recall
|
| 17 |
- f1
|
| 18 |
+
library_name: generic
|
| 19 |
---
|
| 20 |
|
| 21 |
# Binary Change Detection (EO-SAR Fusion)
|
|
|
|
| 26 |
The architecture uses dual weight-shared **ResNet-34** encoders to extract multi-modal features from pre-event RGB (EO) and post-event grayscale (SAR) images. Feature differences are fused via skip connections into a UNet decoder.
|
| 27 |
|
| 28 |
- **Encoder:** ResNet-34 (Pre-trained on ImageNet)
|
| 29 |
+
- **Weights Format:** Safetensors (Modern, fast, and secure)
|
| 30 |
- **Loss Function:** Combined Focal Loss + Dice Loss (Optimized for 58:1 class imbalance)
|
| 31 |
- **Input Resolution:** 256x256
|
| 32 |
|
|
|
|
| 41 |
|
| 42 |
## How to Use
|
| 43 |
The model expects a 3-channel EO image (Pre-event) and a 1-channel SAR image (Post-event) as inputs.
|
| 44 |
+
|
| 45 |
+
```python
|
| 46 |
+
from safetensors.torch import load_file
|
| 47 |
+
# from src.model import SiameseUNet
|
| 48 |
+
|
| 49 |
+
# model = SiameseUNet()
|
| 50 |
+
# weights = load_file("model.safetensors")
|
| 51 |
+
# model.load_state_dict(weights)
|
| 52 |
+
```
|