rishii100 commited on
Commit
c3a61e3
·
verified ·
1 Parent(s): 7d1c03d

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +12 -0
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
+ ```