Upload 11 files
Browse files- .gitattributes +1 -0
- trained_models/multiclass_results_20260106_125323_trans_unet/config/experiment_config.json +20 -0
- trained_models/multiclass_results_20260106_125323_trans_unet/figures/training_curves.pdf +0 -0
- trained_models/multiclass_results_20260106_125323_trans_unet/figures/training_curves.png +3 -0
- trained_models/multiclass_results_20260106_125323_trans_unet/models/multiclass_model.h5 +3 -0
- trained_models/multiclass_results_20260106_125323_trans_unet/models/training_history.pkl +3 -0
- trained_models/multiclass_results_20260106_125323_trans_unet/predictions/predictions.npy +3 -0
- trained_models/multiclass_results_20260106_125323_trans_unet/predictions/test_masks.npy +3 -0
- trained_models/multiclass_results_20260106_125323_trans_unet/tables/trans_unet_latex_table.tex +13 -0
- trained_models/multiclass_results_20260106_125323_trans_unet/tables/trans_unet_metrics.csv +4 -0
- trained_models/multiclass_results_20260106_125323_trans_unet/tables/trans_unet_metrics.xlsx +0 -0
- trained_models/multiclass_results_20260106_125323_trans_unet/trans_unet_summary.txt +53 -0
.gitattributes
CHANGED
|
@@ -35,3 +35,4 @@ saved_model/**/* 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
|
|
|
|
|
|
| 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
|
| 38 |
+
trained_models/multiclass_results_20260106_125323_trans_unet/figures/training_curves.png filter=lfs diff=lfs merge=lfs -text
|
trained_models/multiclass_results_20260106_125323_trans_unet/config/experiment_config.json
ADDED
|
@@ -0,0 +1,20 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"timestamp": "20260106_125323",
|
| 3 |
+
"model_name": "trans_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_125323_trans_unet/figures/training_curves.pdf
ADDED
|
Binary file (18.1 kB). View file
|
|
|
trained_models/multiclass_results_20260106_125323_trans_unet/figures/training_curves.png
ADDED
|
Git LFS Details
|
trained_models/multiclass_results_20260106_125323_trans_unet/models/multiclass_model.h5
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:f99d4fe147f0b77ec788c9c8b20b8ab7e8004a63ad99301a9b5489592ce644f6
|
| 3 |
+
size 468835360
|
trained_models/multiclass_results_20260106_125323_trans_unet/models/training_history.pkl
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:df38dbc3d6722b627cad3cf2a1783e2efb877508df66ccb4a26303de230f9ba8
|
| 3 |
+
size 3689
|
trained_models/multiclass_results_20260106_125323_trans_unet/predictions/predictions.npy
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:51dc73d68d63dbe9d502b3693ce92682c0b1e670d6566c1f1a0309533e4e7b16
|
| 3 |
+
size 41943168
|
trained_models/multiclass_results_20260106_125323_trans_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_125323_trans_unet/tables/trans_unet_latex_table.tex
ADDED
|
@@ -0,0 +1,13 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
\begin{table}
|
| 2 |
+
\caption{Performance metrics for trans_unet model}
|
| 3 |
+
\label{tab:trans_unet_metrics}
|
| 4 |
+
\begin{tabular}{lrrrrr}
|
| 5 |
+
\toprule
|
| 6 |
+
Class & Precision & Recall & Dice & IoU & HD95 \\
|
| 7 |
+
\midrule
|
| 8 |
+
Ventricles & 0.9133 & 0.8948 & 0.9039 & 0.8247 & 8.8822 \\
|
| 9 |
+
Abnormal WMH & 0.8810 & 0.6412 & 0.7422 & 0.5901 & 24.8415 \\
|
| 10 |
+
Overall & 0.8972 & 0.7680 & 0.8231 & 0.7074 & 16.8619 \\
|
| 11 |
+
\bottomrule
|
| 12 |
+
\end{tabular}
|
| 13 |
+
\end{table}
|
trained_models/multiclass_results_20260106_125323_trans_unet/tables/trans_unet_metrics.csv
ADDED
|
@@ -0,0 +1,4 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
Class,Precision,Recall,Dice,IoU,HD95
|
| 2 |
+
Ventricles,0.9132987910189982,0.8947762817568711,0.9039426610837258,0.8247220516204834,8.88221087063197
|
| 3 |
+
Abnormal WMH,0.8810394610202117,0.6411711143797717,0.7422061864133375,0.5900857448577881,24.841499838840388
|
| 4 |
+
Overall,0.897169126019605,0.7679736980683214,0.8230744237485317,0.7074038982391357,16.86185535473618
|
trained_models/multiclass_results_20260106_125323_trans_unet/tables/trans_unet_metrics.xlsx
ADDED
|
Binary file (5.25 kB). View file
|
|
|
trained_models/multiclass_results_20260106_125323_trans_unet/trans_unet_summary.txt
ADDED
|
@@ -0,0 +1,53 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
|
| 2 |
+
MULTI-CLASS WMH AND VENTRICLES SEGMENTATION RESULTS
|
| 3 |
+
===================================================
|
| 4 |
+
Experiment Timestamp: 20260106_153758
|
| 5 |
+
Model Architecture: TRANS_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: TRANS_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.9133 | 0.8810 | 0.8972
|
| 27 |
+
Recall | 0.8948 | 0.6412 | 0.7680
|
| 28 |
+
Dice Coefficient | 0.9039 | 0.7422 | 0.8231
|
| 29 |
+
IoU | 0.8247 | 0.5901 | 0.7074
|
| 30 |
+
HD95 (pixels) | 8.8822 | 24.8415 | 16.8619
|
| 31 |
+
|
| 32 |
+
KEY FINDINGS:
|
| 33 |
+
-------------
|
| 34 |
+
1. Best performing class: Ventricles
|
| 35 |
+
2. Ventricles Dice: 0.9039
|
| 36 |
+
3. Abnormal WMH Dice: 0.7422
|
| 37 |
+
4. Overall Dice: 0.8231
|
| 38 |
+
|
| 39 |
+
FILES GENERATED:
|
| 40 |
+
----------------
|
| 41 |
+
- Model: multiclass_model.h5
|
| 42 |
+
- Figures: training_curves.png/.pdf
|
| 43 |
+
- Tables: trans_unet_metrics.csv/.xlsx, trans_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
|