Upload 11 files
Browse files- .gitattributes +1 -0
- trained_models/multiclass_results_20260106_121152_attn_unet/attn_unet_summary.txt +53 -0
- trained_models/multiclass_results_20260106_121152_attn_unet/config/experiment_config.json +20 -0
- trained_models/multiclass_results_20260106_121152_attn_unet/figures/training_curves.pdf +0 -0
- trained_models/multiclass_results_20260106_121152_attn_unet/figures/training_curves.png +3 -0
- trained_models/multiclass_results_20260106_121152_attn_unet/models/multiclass_model.h5 +3 -0
- trained_models/multiclass_results_20260106_121152_attn_unet/models/training_history.pkl +3 -0
- trained_models/multiclass_results_20260106_121152_attn_unet/predictions/predictions.npy +3 -0
- trained_models/multiclass_results_20260106_121152_attn_unet/predictions/test_masks.npy +3 -0
- trained_models/multiclass_results_20260106_121152_attn_unet/tables/attn_unet_latex_table.tex +13 -0
- trained_models/multiclass_results_20260106_121152_attn_unet/tables/attn_unet_metrics.csv +4 -0
- trained_models/multiclass_results_20260106_121152_attn_unet/tables/attn_unet_metrics.xlsx +0 -0
.gitattributes
CHANGED
|
@@ -34,3 +34,4 @@ saved_model/**/* 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 |
trained_models/multiclass_results_20260106_114110_unet/figures/training_curves.png 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 |
trained_models/multiclass_results_20260106_114110_unet/figures/training_curves.png filter=lfs diff=lfs merge=lfs -text
|
| 37 |
+
trained_models/multiclass_results_20260106_121152_attn_unet/figures/training_curves.png filter=lfs diff=lfs merge=lfs -text
|
trained_models/multiclass_results_20260106_121152_attn_unet/attn_unet_summary.txt
ADDED
|
@@ -0,0 +1,53 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
|
| 2 |
+
MULTI-CLASS WMH AND VENTRICLES SEGMENTATION RESULTS
|
| 3 |
+
===================================================
|
| 4 |
+
Experiment Timestamp: 20260106_150912
|
| 5 |
+
Model Architecture: ATTN_UNET
|
| 6 |
+
|
| 7 |
+
DATASET INFORMATION:
|
| 8 |
+
--------------------
|
| 9 |
+
Training Images: 1775
|
| 10 |
+
Test Images: 199
|
| 11 |
+
Image Size: (256, 256)
|
| 12 |
+
Classes: Background (0), Ventricles (1), Abnormal WMH (2)
|
| 13 |
+
|
| 14 |
+
METHODOLOGY:
|
| 15 |
+
------------
|
| 16 |
+
Architecture: ATTN_UNET
|
| 17 |
+
Loss Function: weighted_categorical
|
| 18 |
+
Training Epochs: 100
|
| 19 |
+
Batch Size: 8
|
| 20 |
+
Learning Rate: 0.0001
|
| 21 |
+
|
| 22 |
+
PERFORMANCE RESULTS:
|
| 23 |
+
--------------------
|
| 24 |
+
| Ventricles | Abnormal WMH | Overall
|
| 25 |
+
--------------------|---------------|---------------|-------------
|
| 26 |
+
Precision | 0.9117 | 0.8345 | 0.8731
|
| 27 |
+
Recall | 0.9226 | 0.7670 | 0.8448
|
| 28 |
+
Dice Coefficient | 0.9171 | 0.7993 | 0.8582
|
| 29 |
+
IoU | 0.8469 | 0.6657 | 0.7563
|
| 30 |
+
HD95 (pixels) | 3.8515 | 20.2336 | 12.0426
|
| 31 |
+
|
| 32 |
+
KEY FINDINGS:
|
| 33 |
+
-------------
|
| 34 |
+
1. Best performing class: Ventricles
|
| 35 |
+
2. Ventricles Dice: 0.9171
|
| 36 |
+
3. Abnormal WMH Dice: 0.7993
|
| 37 |
+
4. Overall Dice: 0.8582
|
| 38 |
+
|
| 39 |
+
FILES GENERATED:
|
| 40 |
+
----------------
|
| 41 |
+
- Model: multiclass_model.h5
|
| 42 |
+
- Figures: training_curves.png/.pdf
|
| 43 |
+
- Tables: attn_unet_metrics.csv/.xlsx, attn_unet_latex_table.tex
|
| 44 |
+
- Predictions: All test predictions and ground truth data saved
|
| 45 |
+
|
| 46 |
+
PUBLICATION READINESS:
|
| 47 |
+
----------------------
|
| 48 |
+
✓ High-resolution figures (300 DPI, PNG/PDF)
|
| 49 |
+
✓ LaTeX-formatted tables
|
| 50 |
+
✓ Comprehensive per-class metrics
|
| 51 |
+
✓ Post-processing applied
|
| 52 |
+
✓ Reproducible results with saved model
|
| 53 |
+
✓ Professional documentation
|
trained_models/multiclass_results_20260106_121152_attn_unet/config/experiment_config.json
ADDED
|
@@ -0,0 +1,20 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"timestamp": "20260106_121152",
|
| 3 |
+
"model_name": "attn_unet",
|
| 4 |
+
"input_shape": [
|
| 5 |
+
256,
|
| 6 |
+
256,
|
| 7 |
+
1
|
| 8 |
+
],
|
| 9 |
+
"target_size": [
|
| 10 |
+
256,
|
| 11 |
+
256
|
| 12 |
+
],
|
| 13 |
+
"num_classes": 3,
|
| 14 |
+
"epochs": 100,
|
| 15 |
+
"batch_size": 8,
|
| 16 |
+
"learning_rate": 0.0001,
|
| 17 |
+
"validation_split": 0.1,
|
| 18 |
+
"random_state": 42,
|
| 19 |
+
"loss_function": "weighted_categorical"
|
| 20 |
+
}
|
trained_models/multiclass_results_20260106_121152_attn_unet/figures/training_curves.pdf
ADDED
|
Binary file (18.8 kB). View file
|
|
|
trained_models/multiclass_results_20260106_121152_attn_unet/figures/training_curves.png
ADDED
|
Git LFS Details
|
trained_models/multiclass_results_20260106_121152_attn_unet/models/multiclass_model.h5
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:43d68fb94e0add529cdf948f78d4600a7d27163a811a63e9fadbd0bf1c953e4f
|
| 3 |
+
size 376915096
|
trained_models/multiclass_results_20260106_121152_attn_unet/models/training_history.pkl
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:540e63f2fe769f65a3a9ab596709c5a4b2359e401ada9b62d11d97d9bd83f2c2
|
| 3 |
+
size 8279
|
trained_models/multiclass_results_20260106_121152_attn_unet/predictions/predictions.npy
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:6113df9217b2db8ac492039782baefa18f95eb3df8f88c3edf05f59e53e8db84
|
| 3 |
+
size 41943168
|
trained_models/multiclass_results_20260106_121152_attn_unet/predictions/test_masks.npy
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:1810b85e3f3303afdb23cd496d98d42e53dea688e09b74c0e336e661984d1be3
|
| 3 |
+
size 5243008
|
trained_models/multiclass_results_20260106_121152_attn_unet/tables/attn_unet_latex_table.tex
ADDED
|
@@ -0,0 +1,13 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
\begin{table}
|
| 2 |
+
\caption{Performance metrics for attn_unet model}
|
| 3 |
+
\label{tab:attn_unet_metrics}
|
| 4 |
+
\begin{tabular}{lrrrrr}
|
| 5 |
+
\toprule
|
| 6 |
+
Class & Precision & Recall & Dice & IoU & HD95 \\
|
| 7 |
+
\midrule
|
| 8 |
+
Ventricles & 0.9117 & 0.9226 & 0.9171 & 0.8469 & 3.8515 \\
|
| 9 |
+
Abnormal WMH & 0.8345 & 0.7670 & 0.7993 & 0.6657 & 20.2336 \\
|
| 10 |
+
Overall & 0.8731 & 0.8448 & 0.8582 & 0.7563 & 12.0426 \\
|
| 11 |
+
\bottomrule
|
| 12 |
+
\end{tabular}
|
| 13 |
+
\end{table}
|
trained_models/multiclass_results_20260106_121152_attn_unet/tables/attn_unet_metrics.csv
ADDED
|
@@ -0,0 +1,4 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
Class,Precision,Recall,Dice,IoU,HD95
|
| 2 |
+
Ventricles,0.911729574773053,0.9225507022166356,0.9171082194917984,0.8469066023826599,3.851465528006527
|
| 3 |
+
Abnormal WMH,0.8344764517604024,0.7669678503887372,0.7992992445052995,0.6656939387321472,20.23364595811516
|
| 4 |
+
Overall,0.8731030132667277,0.8447592763026863,0.8582037319985489,0.7563002705574036,12.042555743060845
|
trained_models/multiclass_results_20260106_121152_attn_unet/tables/attn_unet_metrics.xlsx
ADDED
|
Binary file (5.25 kB). View file
|
|
|