kreyesp commited on
Commit
7dafc62
·
verified ·
1 Parent(s): c227be5

Add model card

Browse files
Files changed (1) hide show
  1. README.md +60 -0
README.md ADDED
@@ -0,0 +1,60 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: mit
3
+ tags:
4
+ - image-segmentation
5
+ - satellite-imagery
6
+ - cloud-detection
7
+ - cubesat
8
+ - beavercube
9
+ ---
10
+
11
+ # BeaverCube Cloud Segmentation — SmallCloudNet
12
+
13
+ U-Net trained to detect clouds in imagery simulated from the Matrix Vision mvBlueFOX-IGC-200w camera
14
+ on the BeaverCube 2 CubeSat (MIT), using the CloudSEN12-L1C Sentinel-2 dataset.
15
+
16
+ ## Model details
17
+
18
+ | Property | Value |
19
+ |---|---|
20
+ | Architecture | U-Net (SmallCloudNet) |
21
+ | Parameters | 1.86M |
22
+ | Input size (training) | 33×33 px |
23
+ | Input size (inference) | any (fully convolutional) |
24
+ | Classes | clear, thick cloud, thin cloud, shadow |
25
+
26
+ ## Performance
27
+
28
+ | Metric | Value |
29
+ |---|---|
30
+ | Mean IoU | 0.38 |
31
+ | Mean F1 | 0.54 |
32
+ | Accuracy | 64% |
33
+ | Clear IoU | 0.58 |
34
+ | Thick cloud IoU | 0.45 |
35
+ | Shadow IoU | 0.27 |
36
+ | Thin cloud IoU | 0.22 |
37
+
38
+ ## Usage
39
+
40
+ ```python
41
+ import torch
42
+ from model import SmallCloudNet
43
+
44
+ model = SmallCloudNet(in_ch=3, num_classes=4)
45
+ checkpoint = torch.load("best_model.pth", map_location="cpu")
46
+ model.load_state_dict(checkpoint["model_state_dict"])
47
+ model.eval()
48
+
49
+ # img: float32 tensor (1, 3, H, W) normalised to [0, 1]
50
+ with torch.no_grad():
51
+ logits = model(img) # (1, 4, H, W)
52
+ mask = logits.argmax(dim=1) # (1, H, W)
53
+ ```
54
+
55
+ ## Training data
56
+
57
+ CloudSEN12-L1C (Sentinel-2 L1C), preprocessed to simulate BlueFOX GSD (153.75 m) via:
58
+ 1. Gaussian PSF blur (σ = 1.6 px, derived from Kowa 16mm lens Airy disk)
59
+ 2. 15.65× INTER_AREA downsample (512×512 → 33×33)
60
+ 3. Read + shot noise augmentation applied each epoch