Hugging Face's logo

bwittmann
/
octa-unetplusplus

Model card Files
xet
Community
bwittmann commited on
Commit
9cdee89
·
verified ·
1 Parent(s): 646b2c4

Upload weights used to initialize pre-trained models.

Browse files
Files changed (8) hide show
  1. superunet_syn_base_cldice01_moreparams/config.json +116 -0
  2. superunet_syn_base_cldice01_moreparams/events.out.tfevents.1713183961.u20-computegpu-3.1599522.0 +3 -0
  3. superunet_syn_base_cldice01_moreparams/model_best_0.757.pt +3 -0
  4. superunet_syn_base_cldice01_moreparams/model_last.pt +3 -0
  5. superunet_syn_base_moreparams/config.json +116 -0
  6. superunet_syn_base_moreparams/events.out.tfevents.1713203288.u20-computeibmgpu-vesta13.1228735.0 +3 -0
  7. superunet_syn_base_moreparams/model_best_0.763.pt +3 -0
  8. superunet_syn_base_moreparams/model_last.pt +3 -0
superunet_syn_base_cldice01_moreparams/config.json ADDED
@@ -0,0 +1,116 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "sim_data": "/home/bwittm/share/synthetic_cerebral_octa",
3
+ "sim_data_version": "b421",
4
+ "annotations": "/home/bwittm/share/manual_annotations",
5
+ "test_imgs": [
6
+ "m4_0"
7
+ ],
8
+ "val_imgs": [
9
+ "m4_0"
10
+ ],
11
+ "train_on_real": false,
12
+ "train_img": [
13
+ "m4_1",
14
+ "m44_0",
15
+ "m44_1",
16
+ "m78_0",
17
+ "m78_1"
18
+ ],
19
+ "experiment_name": "superunet_syn_base_cldice01_moreparams",
20
+ "device": "cuda:0",
21
+ "store_ckpt": true,
22
+ "epochs": 300,
23
+ "epochs_decay": 50,
24
+ "val_interval": 1,
25
+ "itera_pro_epoch": 50000,
26
+ "seed": 0,
27
+ "batch_size": 32,
28
+ "num_workers": 16,
29
+ "sliding_window_overlap": 0.5,
30
+ "sliding_window_mode": "constant",
31
+ "threshold": 0.5,
32
+ "seg": {
33
+ "lr": "1e-5",
34
+ "channels": [
35
+ 64,
36
+ 128,
37
+ 256,
38
+ 512,
39
+ 1024
40
+ ],
41
+ "strides": [
42
+ 2,
43
+ 2,
44
+ 2,
45
+ 2
46
+ ],
47
+ "activation": "LEAKYRELU",
48
+ "normalization": "BATCH",
49
+ "dropout": 0.3,
50
+ "max_norm_clip": 1.0,
51
+ "cldice": true,
52
+ "iter": 15,
53
+ "alpha": 0.1,
54
+ "smooth": 1
55
+ },
56
+ "augmentation": {
57
+ "patch_size": [
58
+ 64,
59
+ 64,
60
+ 64
61
+ ],
62
+ "use_augmentation": true,
63
+ "p_gaussian_noise": 0.5,
64
+ "p_gaussian_smooth": 0.5,
65
+ "p_intensity_scale": 0,
66
+ "p_intensity_shift": 0,
67
+ "p_adjust_contrast": 0,
68
+ "p_histogram_shift": 0.5,
69
+ "p_rotate": 0.5,
70
+ "p_zoom": 0.5,
71
+ "p_shear": 0,
72
+ "p_flip": [
73
+ 0,
74
+ 0.5,
75
+ 0.5
76
+ ],
77
+ "gaussian_noise_mean": 0.0,
78
+ "gaussian_noise_std": 0.01,
79
+ "gaussian_smooth_sigma": [
80
+ 0.0,
81
+ 0.5
82
+ ],
83
+ "intensity_scale_factors": [
84
+ -0.5,
85
+ 1.2
86
+ ],
87
+ "intensity_shift_offsets": [
88
+ 0.05,
89
+ 0.2
90
+ ],
91
+ "adjust_contrast_gamma": [
92
+ 0.7,
93
+ 1.3
94
+ ],
95
+ "control_points": [
96
+ 5,
97
+ 10
98
+ ],
99
+ "rotation": [
100
+ -90,
101
+ 90
102
+ ],
103
+ "min_zoom": 1.0,
104
+ "max_zoom": 1.2,
105
+ "shear_range": [
106
+ 0.1,
107
+ 0.1,
108
+ 0.1
109
+ ]
110
+ },
111
+ "git_commit_hash": "7a8f7f7",
112
+ "python_version": "3.8.19 | packaged by conda-forge | (default, Mar 20 2024, 12:47:35) ",
113
+ "gcc_version": "[GCC 12.3.0]",
114
+ "pytorch_version": "1.13.1+cu116",
115
+ "host_name": "u20-computegpu-3"
116
+ }
superunet_syn_base_cldice01_moreparams/events.out.tfevents.1713183961.u20-computegpu-3.1599522.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ae9d94b64e50ddf4d68290aa3b9b3deb202a865b42ab97c5b19c180818d6f6de
3
+ size 415795
superunet_syn_base_cldice01_moreparams/model_best_0.757.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cc3f754b23fdeca0759571db2e2bb4a88bca92f01fa3222e85a5a32cd68ff3f4
3
+ size 379775475
superunet_syn_base_cldice01_moreparams/model_last.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4f9c74c65c7e5386c89a5b87d6253c94d2af13d2d3b34bd5c62c60b68ee54a47
3
+ size 379774503
superunet_syn_base_moreparams/config.json ADDED
@@ -0,0 +1,116 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "sim_data": "/home/bwittm/share/synthetic_cerebral_octa",
3
+ "sim_data_version": "b421",
4
+ "annotations": "/home/bwittm/share/manual_annotations",
5
+ "test_imgs": [
6
+ "m4_0"
7
+ ],
8
+ "val_imgs": [
9
+ "m4_0"
10
+ ],
11
+ "train_on_real": false,
12
+ "train_img": [
13
+ "m4_1",
14
+ "m44_0",
15
+ "m44_1",
16
+ "m78_0",
17
+ "m78_1"
18
+ ],
19
+ "experiment_name": "superunet_syn_base_moreparams",
20
+ "device": "cuda:0",
21
+ "store_ckpt": true,
22
+ "epochs": 300,
23
+ "epochs_decay": 50,
24
+ "val_interval": 1,
25
+ "itera_pro_epoch": 50000,
26
+ "seed": 0,
27
+ "batch_size": 32,
28
+ "num_workers": 16,
29
+ "sliding_window_overlap": 0.5,
30
+ "sliding_window_mode": "constant",
31
+ "threshold": 0.5,
32
+ "seg": {
33
+ "lr": "1e-5",
34
+ "channels": [
35
+ 64,
36
+ 128,
37
+ 256,
38
+ 512,
39
+ 1024
40
+ ],
41
+ "strides": [
42
+ 2,
43
+ 2,
44
+ 2,
45
+ 2
46
+ ],
47
+ "activation": "LEAKYRELU",
48
+ "normalization": "BATCH",
49
+ "dropout": 0.3,
50
+ "max_norm_clip": 1.0,
51
+ "cldice": false,
52
+ "iter": 15,
53
+ "alpha": 0.1,
54
+ "smooth": 1
55
+ },
56
+ "augmentation": {
57
+ "patch_size": [
58
+ 64,
59
+ 64,
60
+ 64
61
+ ],
62
+ "use_augmentation": true,
63
+ "p_gaussian_noise": 0.5,
64
+ "p_gaussian_smooth": 0.5,
65
+ "p_intensity_scale": 0,
66
+ "p_intensity_shift": 0,
67
+ "p_adjust_contrast": 0,
68
+ "p_histogram_shift": 0.5,
69
+ "p_rotate": 0.5,
70
+ "p_zoom": 0.5,
71
+ "p_shear": 0,
72
+ "p_flip": [
73
+ 0,
74
+ 0.5,
75
+ 0.5
76
+ ],
77
+ "gaussian_noise_mean": 0.0,
78
+ "gaussian_noise_std": 0.01,
79
+ "gaussian_smooth_sigma": [
80
+ 0.0,
81
+ 0.5
82
+ ],
83
+ "intensity_scale_factors": [
84
+ -0.5,
85
+ 1.2
86
+ ],
87
+ "intensity_shift_offsets": [
88
+ 0.05,
89
+ 0.2
90
+ ],
91
+ "adjust_contrast_gamma": [
92
+ 0.7,
93
+ 1.3
94
+ ],
95
+ "control_points": [
96
+ 5,
97
+ 10
98
+ ],
99
+ "rotation": [
100
+ -90,
101
+ 90
102
+ ],
103
+ "min_zoom": 1.0,
104
+ "max_zoom": 1.2,
105
+ "shear_range": [
106
+ 0.1,
107
+ 0.1,
108
+ 0.1
109
+ ]
110
+ },
111
+ "git_commit_hash": "7a8f7f7",
112
+ "python_version": "3.8.19 | packaged by conda-forge | (default, Mar 20 2024, 12:47:35) ",
113
+ "gcc_version": "[GCC 12.3.0]",
114
+ "pytorch_version": "1.13.1+cu116",
115
+ "host_name": "u20-computeibmgpu-vesta13"
116
+ }
superunet_syn_base_moreparams/events.out.tfevents.1713203288.u20-computeibmgpu-vesta13.1228735.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8984e8a22d8081d3220f6bbd428e6aa21a023f7a17eedcf817e60488cbd9b227
3
+ size 578503
superunet_syn_base_moreparams/model_best_0.763.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:49f81d2d0c57169b29f2fd1e7ea0227a5bafa0c5eac7f0611498c0c9b2960d12
3
+ size 379775411
superunet_syn_base_moreparams/model_last.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:670787f0634062eee6b980d9cb6bcf75df702d9e4998e601cafb402152041b68
3
+ size 379774439