Log hyper-parameters, metrics plots, and model state for unet_run_2026-06-23_13-16
Browse files- .gitattributes +1 -0
- README.md +45 -0
- checkpoint-12474/model.safetensors +3 -0
- checkpoint-12474/optimizer.pt +3 -0
- checkpoint-12474/rng_state.pth +3 -0
- checkpoint-12474/scheduler.pt +3 -0
- checkpoint-12474/trainer_state.json +0 -0
- checkpoint-12474/training_args.bin +3 -0
- checkpoint-13134/model.safetensors +3 -0
- checkpoint-13134/optimizer.pt +3 -0
- checkpoint-13134/rng_state.pth +3 -0
- checkpoint-13134/scheduler.pt +3 -0
- checkpoint-13134/trainer_state.json +0 -0
- checkpoint-13134/training_args.bin +3 -0
- checkpoint-13200/model.safetensors +3 -0
- checkpoint-13200/optimizer.pt +3 -0
- checkpoint-13200/rng_state.pth +3 -0
- checkpoint-13200/scheduler.pt +3 -0
- checkpoint-13200/trainer_state.json +0 -0
- checkpoint-13200/training_args.bin +3 -0
- config.json +8 -0
- performance_curves.png +3 -0
- pytorch_model.bin +2 -2
- training_metrics.csv +0 -0
.gitattributes
CHANGED
|
@@ -33,3 +33,4 @@ 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 |
+
performance_curves.png filter=lfs diff=lfs merge=lfs -text
|
README.md
ADDED
|
@@ -0,0 +1,45 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
language: en
|
| 3 |
+
license: mit
|
| 4 |
+
tags:
|
| 5 |
+
- semantic-segmentation
|
| 6 |
+
- pytorch
|
| 7 |
+
- unet
|
| 8 |
+
- resnet34
|
| 9 |
+
- agricultural-cv
|
| 10 |
+
model-index:
|
| 11 |
+
- name: unet_run_2026-06-23_13-16
|
| 12 |
+
results:
|
| 13 |
+
- task:
|
| 14 |
+
type: semantic-segmentation
|
| 15 |
+
name: Corm Segmentation
|
| 16 |
+
metrics:
|
| 17 |
+
- type: mean_iou
|
| 18 |
+
value: 0.9293187884547999
|
| 19 |
+
name: Best Val Mean IoU
|
| 20 |
+
- type: mean_accuracy
|
| 21 |
+
value: 0.9665135754918103
|
| 22 |
+
name: Best Val Mean Accuracy
|
| 23 |
+
---
|
| 24 |
+
|
| 25 |
+
# UNet Corm & Damage Semantic Segmentation Model
|
| 26 |
+
|
| 27 |
+
This repository contains the weights, performance logs, and hyperparameters for experiment `unet_run_2026-06-23_13-16`.
|
| 28 |
+
|
| 29 |
+
## Model Hyperparameters
|
| 30 |
+
- **Architecture Type:** UNet (SMP wrapper)
|
| 31 |
+
- **Backbone Encoder:** `UNet`
|
| 32 |
+
- **Pretrained Weights:** `imagenet`
|
| 33 |
+
- **Input Channels:** 3
|
| 34 |
+
- **Number of Classes:** 3 (0: background, 1: damage, 2: root)
|
| 35 |
+
- **Training Resolution:** `256x256`
|
| 36 |
+
|
| 37 |
+
## Training Configurations
|
| 38 |
+
- **Batch Size:** 8
|
| 39 |
+
- **Optimizers / Lr:** AdamW / 0.0001
|
| 40 |
+
- **Maximum Epochs:** 200
|
| 41 |
+
- **Seed Configuration:** 42
|
| 42 |
+
|
| 43 |
+
## Metrics Curves
|
| 44 |
+
Below are the training performance plots generated for this run:
|
| 45 |
+

|
checkpoint-12474/model.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:1b725bd9edec2e7c444c750e77aa2c5a2384312c268c2d1be7dbbeb69d2fa51c
|
| 3 |
+
size 97852220
|
checkpoint-12474/optimizer.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:52053157bfe09611c2231de16eacb1cde2fd881b61b4bb7fce07d393065cbce9
|
| 3 |
+
size 195611002
|
checkpoint-12474/rng_state.pth
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:20e02127f3f897ff3f2b45fcce32bf8c197ec4e8a907246bb0ef6176e4499362
|
| 3 |
+
size 14244
|
checkpoint-12474/scheduler.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:bed679b273ab4b37f55f8b35294ec40bc4714a9d7415e96e96e4b0d132bbe02e
|
| 3 |
+
size 1064
|
checkpoint-12474/trainer_state.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
checkpoint-12474/training_args.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:2c2a298fe86596eddfae49fcfab10c90acf7f9ec0881f052038406bf6efa56f8
|
| 3 |
+
size 5304
|
checkpoint-13134/model.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:f9c857874cb8d75c0657b6aca692cd361a368947c0ba05d9bb55fe52ff31f8bc
|
| 3 |
+
size 97852220
|
checkpoint-13134/optimizer.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:ff8d5bf82002bab767403ed5b8b15069f9ab1462049d16f471d1837f3689d55b
|
| 3 |
+
size 195611002
|
checkpoint-13134/rng_state.pth
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:c77813d9c400a219ef282d839ccdbb0ae28a23b00491136a2e3e6c163f11bf95
|
| 3 |
+
size 14244
|
checkpoint-13134/scheduler.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:4d70d82ac190b04b03e523cf82197c4386631af0c9a2d740fb2ef969cfc9e81b
|
| 3 |
+
size 1064
|
checkpoint-13134/trainer_state.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
checkpoint-13134/training_args.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:2c2a298fe86596eddfae49fcfab10c90acf7f9ec0881f052038406bf6efa56f8
|
| 3 |
+
size 5304
|
checkpoint-13200/model.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:010f1e3f9b57fecbccefc2d078597ca2b6a5b8a34fe3dfe12622fa17aaa4f7cb
|
| 3 |
+
size 97852220
|
checkpoint-13200/optimizer.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:85ecf2efce5fd0ad0f82639f598c9a0fbc429b82d1d049776007d88296f517f3
|
| 3 |
+
size 195611002
|
checkpoint-13200/rng_state.pth
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:975f111019f082e813964463b3f2d36f0b84d420a867a00b1d777dcd04304e1c
|
| 3 |
+
size 14244
|
checkpoint-13200/scheduler.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:cac21f4ad30262035e7514efacb0c83e98c4a92c1b231bb7f5b6519c22773d20
|
| 3 |
+
size 1064
|
checkpoint-13200/trainer_state.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
checkpoint-13200/training_args.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:2c2a298fe86596eddfae49fcfab10c90acf7f9ec0881f052038406bf6efa56f8
|
| 3 |
+
size 5304
|
config.json
ADDED
|
@@ -0,0 +1,8 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"model_type": "UNet",
|
| 3 |
+
"encoder_name": "resnet34",
|
| 4 |
+
"encoder_weights": "imagenet",
|
| 5 |
+
"in_channels": 3,
|
| 6 |
+
"classes": 3,
|
| 7 |
+
"input_resolution": "256x256"
|
| 8 |
+
}
|
performance_curves.png
ADDED
|
Git LFS Details
|
pytorch_model.bin
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
-
size
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:9ad52735fe47a835c5df2158d605c12b76f474c893ab7b99b2f4db3d6c5d5ed5
|
| 3 |
+
size 97924706
|
training_metrics.csv
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|