Bawil commited on
Commit
85f397a
·
verified ·
1 Parent(s): 60dfc0c

Upload 11 files

Browse files
.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

  • SHA256: 22e473a8c277b4cd69cebd55e25f6a41eda46d58b6fb26d30623a5a9a16e4b44
  • Pointer size: 131 Bytes
  • Size of remote file: 207 kB
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