Update README.md
Browse files
README.md
CHANGED
@@ -108,7 +108,7 @@ pipeline_tag: image-segmentation
|
|
108 |
- **Compute infrastructure:**
|
109 |
- software: python, pytorch-lightning
|
110 |
- hardware: HPC/AI resources provided by GENCI-IDRIS
|
111 |
-
- **License:** :
|
112 |
|
113 |
---
|
114 |
|
@@ -197,24 +197,25 @@ Statistics of the TRAIN+VALIDATION set :
|
|
197 |
|
198 |
#### Training Hyperparameters
|
199 |
|
200 |
-
|
201 |
-
|
202 |
-
|
203 |
-
|
204 |
-
|
205 |
-
|
206 |
-
|
207 |
-
|
208 |
-
|
209 |
-
|
210 |
-
|
211 |
-
|
212 |
-
|
213 |
-
|
214 |
-
|
215 |
-
|
216 |
-
|
217 |
-
|
|
|
218 |
|
219 |
#### Speeds, Sizes, Times
|
220 |
|
|
|
108 |
- **Compute infrastructure:**
|
109 |
- software: python, pytorch-lightning
|
110 |
- hardware: HPC/AI resources provided by GENCI-IDRIS
|
111 |
+
- **License:** : Etalab 2.0
|
112 |
|
113 |
---
|
114 |
|
|
|
197 |
|
198 |
#### Training Hyperparameters
|
199 |
|
200 |
+
```yaml
|
201 |
+
- Model architecture: Unet #(implementation from the [Segmentation Models Pytorch library](https://segmentation-modelspytorch.readthedocs.io/en/latest/docs/api.html#unet))
|
202 |
+
- Encoder : Resnet-34 pre-trained with ImageNet
|
203 |
+
- Augmentation :
|
204 |
+
- VerticalFlip(p=0.5)
|
205 |
+
- HorizontalFlip(p=0.5)
|
206 |
+
- RandomRotate90(p=0.5)
|
207 |
+
- Input normalization (mean=0 | std=1):
|
208 |
+
- norm_means: [105.08, 110.87, 101.82, 106.38, 53.26]
|
209 |
+
- norm_stds: [52.17, 45.38, 44, 39.69, 79.3]
|
210 |
+
- Seed: 2022
|
211 |
+
- Batch size: 10
|
212 |
+
- Number of epochs : 200
|
213 |
+
- Early stopping : patience 30 and val_loss as monitor criterium
|
214 |
+
- Optimizer : SGD
|
215 |
+
- Schaeduler : mode = "min", factor = 0.5, patience = 10, cooldown = 4, min_lr = 1e-7
|
216 |
+
- Learning rate : 0.02
|
217 |
+
- Class Weights : [1-building: 1.0 , 2-pervious surface: 1.0 , 3-impervious surface: 1.0 , 4-bare soil: 1.0 , 5-water: 1.0 , 6-coniferous: 1.0 , 7-deciduous: 1.0 , 8-brushwood: 1.0 , 9-vineyard: 1.0 , 10-herbaceous vegetation: 1.0 , 11-agricultural land: 1.0 , 12-plowed land: 1.0 , 13-swimming_pool: 1.0 , 14-snow: 1.0 , 15-clear cut: 0.0 , 16-mixed: 0.0 , 17-ligneous: 0.0 , 18-greenhouse: 1.0 , 19-other: 0.0]
|
218 |
+
```
|
219 |
|
220 |
#### Speeds, Sizes, Times
|
221 |
|