Jesteban247 commited on
Commit
251ac28
·
verified ·
1 Parent(s): 87bd443

Upload 6 files

Browse files
.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
+ Pred_2d.png filter=lfs diff=lfs merge=lfs -text
37
+ Prediction_3D.png filter=lfs diff=lfs merge=lfs -text
Pred_2d.png ADDED

Git LFS Details

  • SHA256: e54aa12948b77a8995bf943801fb416473c812b44cc703243c94b0e974d58725
  • Pointer size: 131 Bytes
  • Size of remote file: 164 kB
Prediction_3D.png ADDED

Git LFS Details

  • SHA256: 8037fa681c60ff11804581e1bfa96c49006c4cfa50f249791386439f73d9bdca
  • Pointer size: 131 Bytes
  • Size of remote file: 132 kB
config.txt ADDED
@@ -0,0 +1,41 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ BraTS 3D Segmentation Training Configuration with MONAI + Accelerate
2
+ ==================================================
3
+
4
+ Training Date: 2025-10-09 06:01:59
5
+
6
+ Data Configuration:
7
+ Data Directory: Data/BraTS2020_TrainingData/MICCAI_BraTS2020_TrainingData
8
+ ROI Size (Inference): (128, 128, 128)
9
+ Number of Classes: 3
10
+ Train Split: 80.0%
11
+ Val Split: 5.0%
12
+ Test Split: 15.0%
13
+ No Augmentation (Original Images Only)
14
+
15
+ Model Configuration:
16
+ Model: SegResNet
17
+ Input Channels: 4
18
+ Output Channels: 3
19
+ Initial Filters: 16
20
+
21
+ Training Configuration:
22
+ Framework: Accelerate
23
+ Mixed Precision: fp16
24
+ Optimizer: Adam
25
+ Batch Size (per GPU): 2
26
+ Batch Size (Val/Test per GPU): 1
27
+ Epochs: 150
28
+ Learning Rate: 0.0001
29
+ Weight Decay: 1e-05
30
+ Validation Interval: 5
31
+ Use Caching: False
32
+
33
+ Hardware Configuration:
34
+ Number of Workers per GPU: 8
35
+ Pin Memory: True
36
+ Prefetch Factor: 4
37
+
38
+ Class Names (Multi-Label):
39
+ Channel 0: Tumor Core (TC)
40
+ Channel 1: Whole Tumor (WT)
41
+ Channel 2: Enhancing Tumor (ET)
model_final_20251009_122935.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3c961ac9068308a6a0a86d31a5021f282d97f6581ffd9163a4fce2284f2186c2
3
+ size 18874395
summary.json ADDED
@@ -0,0 +1,76 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "total_epochs": 150,
3
+ "total_time_seconds": 23254.743891716003,
4
+ "total_time_hours": 6.459651081032223,
5
+ "batch_size_per_gpu": 2,
6
+ "num_processes": 2,
7
+ "mixed_precision": "fp16",
8
+ "final_train_loss": 0.11351284384727478,
9
+ "final_train_metrics": {
10
+ "precision": [
11
+ 0.9284262657165527,
12
+ 0.9326722621917725,
13
+ 0.8191995024681091
14
+ ],
15
+ "recall": [
16
+ 0.9270820021629333,
17
+ 0.9292428493499756,
18
+ 0.8744466304779053
19
+ ],
20
+ "accuracy": [
21
+ 0.9996001720428467,
22
+ 0.9991011619567871,
23
+ 0.9927959442138672
24
+ ],
25
+ "dice": [
26
+ 0.9271973371505737,
27
+ 0.9303600788116455,
28
+ 0.8458120822906494
29
+ ]
30
+ },
31
+ "final_val_metrics": {
32
+ "precision": [
33
+ 0.8964494466781616,
34
+ 0.9273992776870728,
35
+ 0.7346808910369873
36
+ ],
37
+ "recall": [
38
+ 0.8001397848129272,
39
+ 0.8460437655448914,
40
+ 0.7222132682800293
41
+ ],
42
+ "accuracy": [
43
+ 0.9991061687469482,
44
+ 0.9987545013427734,
45
+ 0.8885079026222229
46
+ ],
47
+ "dice": [
48
+ 0.8246815204620361,
49
+ 0.8800930976867676,
50
+ 0.7170625329017639
51
+ ]
52
+ },
53
+ "final_test_loss": 0.1920088231563568,
54
+ "final_test_metrics": {
55
+ "precision": [
56
+ 0.8555397391319275,
57
+ 0.93755042552948,
58
+ 0.7489745616912842
59
+ ],
60
+ "recall": [
61
+ 0.7621530294418335,
62
+ 0.838932991027832,
63
+ 0.7132463455200195
64
+ ],
65
+ "accuracy": [
66
+ 0.9988489151000977,
67
+ 0.9985865354537964,
68
+ 0.9637566804885864
69
+ ],
70
+ "dice": [
71
+ 0.7748545408248901,
72
+ 0.8739949464797974,
73
+ 0.7190028429031372
74
+ ]
75
+ }
76
+ }
test_metrics.csv ADDED
@@ -0,0 +1,2 @@
 
 
 
1
+ epoch,loss,avg_dice,avg_precision,avg_recall,avg_accuracy,Tumor Core (TC)_dice,Tumor Core (TC)_precision,Tumor Core (TC)_recall,Tumor Core (TC)_accuracy,Whole Tumor (WT)_dice,Whole Tumor (WT)_precision,Whole Tumor (WT)_recall,Whole Tumor (WT)_accuracy,Enhancing Tumor (ET)_dice,Enhancing Tumor (ET)_precision,Enhancing Tumor (ET)_recall,Enhancing Tumor (ET)_accuracy,epoch_time_sec
2
+ 150,0.192009,0.789284,0.847355,0.771444,0.987064,0.774855,0.855540,0.762153,0.998849,0.873995,0.937550,0.838933,0.998587,0.719003,0.748975,0.713246,0.963757,0.00