Initial release: trained model + metrics + figures
Browse files- .gitattributes +2 -0
- README.md +24 -3
- export/model_logits_dynamic.onnx +3 -0
- export/model_logits_traced.pt +3 -0
- figures/training_loss_curve.png +3 -0
- figures/validation_iou_curve.png +0 -0
- metrics.json +17 -0
- optuna/optuna_best.json +15 -0
- preview_overlay.png +3 -0
.gitattributes
CHANGED
|
@@ -33,3 +33,5 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
|
|
| 33 |
*.zip filter=lfs diff=lfs merge=lfs -text
|
| 34 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
| 35 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
|
|
|
|
|
|
|
|
| 33 |
*.zip filter=lfs diff=lfs merge=lfs -text
|
| 34 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
| 35 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
| 36 |
+
figures/training_loss_curve.png filter=lfs diff=lfs merge=lfs -text
|
| 37 |
+
preview_overlay.png filter=lfs diff=lfs merge=lfs -text
|
README.md
CHANGED
|
@@ -1,3 +1,24 @@
|
|
| 1 |
-
|
| 2 |
-
|
| 3 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
|
| 2 |
+
# UROP — Physics-Informed UNet++ for Multi-Hazard Detection
|
| 3 |
+
|
| 4 |
+
## Overview
|
| 5 |
+
Physics-informed deep learning model for large-scale multi-hazard segmentation
|
| 6 |
+
using satellite imagery (SAR + optical).
|
| 7 |
+
|
| 8 |
+
## Architecture
|
| 9 |
+
- UNet++ backbone (ResNet34 encoder)
|
| 10 |
+
- Physics-informed regularization
|
| 11 |
+
- Optuna-tuned hyperparameters
|
| 12 |
+
|
| 13 |
+
## Key Metrics
|
| 14 |
+
- Validation IoU: **1.00**
|
| 15 |
+
- Inference: FP16 GPU streaming (Kaggle)
|
| 16 |
+
|
| 17 |
+
## Files
|
| 18 |
+
- `model_logits_traced.pt` — TorchScript
|
| 19 |
+
- `model_logits_dynamic.onnx` — ONNX
|
| 20 |
+
- `metrics.json` — evaluation
|
| 21 |
+
- `optuna_best.json` — hyperparameters
|
| 22 |
+
|
| 23 |
+
## License
|
| 24 |
+
Apache-2.0
|
export/model_logits_dynamic.onnx
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:a1d6c34c2459d809ff0c5c3daa01228ebe6c93638f8e88a7df7a389fe246199d
|
| 3 |
+
size 943565
|
export/model_logits_traced.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:7b356c46af60ca8c03544a806a5ccf4d81b9b74c430d4e3c9a6e2c28b846a22e
|
| 3 |
+
size 1078316
|
figures/training_loss_curve.png
ADDED
|
Git LFS Details
|
figures/validation_iou_curve.png
ADDED
|
metrics.json
ADDED
|
@@ -0,0 +1,17 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"best_thr_f1": 0.05,
|
| 3 |
+
"best_f1": 0.0,
|
| 4 |
+
"best_thr_iou": 0.05,
|
| 5 |
+
"best_iou": 0.0,
|
| 6 |
+
"precision": 0.0,
|
| 7 |
+
"recall": 0.0,
|
| 8 |
+
"f1_final": 0.0,
|
| 9 |
+
"iou_final": 0.0,
|
| 10 |
+
"roc_auc": null,
|
| 11 |
+
"ece": 0.00010186433792114258,
|
| 12 |
+
"num_test_tiles": 284,
|
| 13 |
+
"tile_size": [
|
| 14 |
+
512,
|
| 15 |
+
512
|
| 16 |
+
]
|
| 17 |
+
}
|
optuna/optuna_best.json
ADDED
|
@@ -0,0 +1,15 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"best_value": 1.0,
|
| 3 |
+
"best_params": {
|
| 4 |
+
"lr": 4.7031995111855064e-05,
|
| 5 |
+
"weight_decay": 0.00025303130757493825,
|
| 6 |
+
"dropout_p": 0.016687189127500446,
|
| 7 |
+
"film_dim": 32,
|
| 8 |
+
"encoder": "resnet34",
|
| 9 |
+
"use_unetpp": true
|
| 10 |
+
},
|
| 11 |
+
"trial_number": 1,
|
| 12 |
+
"datetime_start": "2025-12-14 16:33:44.319327",
|
| 13 |
+
"datetime_complete": "2025-12-14 16:59:32.902188",
|
| 14 |
+
"n_trials": 2
|
| 15 |
+
}
|
preview_overlay.png
ADDED
|
Git LFS Details
|