Hugging Face's logo

bwittmann
/
octa-unet

Model card Files Community
bwittmann commited on
Commit
ed50c7c
·
verified ·
1 Parent(s): 5058bda

Upload 12 files

Browse files
Files changed (13) hide show
  1. .gitattributes +6 -0
  2. manual_annotated/config.json +80 -0
  3. manual_annotated/model.pt +3 -0
  4. manual_annotated_synthetic/config.json +80 -0
  5. manual_annotated_synthetic/model.pt +3 -0
  6. splits/test_data.nii +3 -0
  7. splits/test_label.nii +3 -0
  8. splits/train_data.nii +3 -0
  9. splits/train_label.nii +3 -0
  10. splits/val_data.nii +3 -0
  11. splits/val_label.nii +3 -0
  12. synthetic/config.json +80 -0
  13. synthetic/model.pt +3 -0
.gitattributes CHANGED
@@ -33,3 +33,9 @@ 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
+ splits/test_data.nii filter=lfs diff=lfs merge=lfs -text
37
+ splits/test_label.nii filter=lfs diff=lfs merge=lfs -text
38
+ splits/train_data.nii filter=lfs diff=lfs merge=lfs -text
39
+ splits/train_label.nii filter=lfs diff=lfs merge=lfs -text
40
+ splits/val_data.nii filter=lfs diff=lfs merge=lfs -text
41
+ splits/val_label.nii filter=lfs diff=lfs merge=lfs -text
manual_annotated/config.json ADDED
@@ -0,0 +1,80 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "batch_size": 32,
3
+ "sliding_window_mode": "constant",
4
+ "models": {
5
+ "lr": "1e-5",
6
+ "wd": 0,
7
+ "dropout": 0.0,
8
+ "model": "unet",
9
+ "unet": {
10
+ "channels": [
11
+ 32,
12
+ 64,
13
+ 128,
14
+ 256
15
+ ],
16
+ "strides": [
17
+ 2,
18
+ 2,
19
+ 2
20
+ ],
21
+ "activation": "LEAKYRELU",
22
+ "normalization": "BATCH"
23
+ }
24
+ },
25
+ "augmentation": {
26
+ "patch_size": [
27
+ 64,
28
+ 64,
29
+ 64
30
+ ],
31
+ "use_augmentation": true,
32
+ "p_gaussian_noise": 0.5,
33
+ "p_gaussian_smooth": 0.5,
34
+ "p_intensity_scale": 0,
35
+ "p_intensity_shift": 0,
36
+ "p_adjust_contrast": 0,
37
+ "p_histogram_shift": 0.5,
38
+ "p_rotate": 0.5,
39
+ "p_zoom": 0.5,
40
+ "p_shear": 0,
41
+ "p_flip": [
42
+ 0,
43
+ 0.5,
44
+ 0.5
45
+ ],
46
+ "gaussian_noise_mean": 0.0,
47
+ "gaussian_noise_std": 0.01,
48
+ "gaussian_smooth_sigma": [
49
+ 0.0,
50
+ 0.5
51
+ ],
52
+ "intensity_scale_factors": [
53
+ -0.5,
54
+ 1.2
55
+ ],
56
+ "intensity_shift_offsets": [
57
+ 0.05,
58
+ 0.2
59
+ ],
60
+ "adjust_contrast_gamma": [
61
+ 0.7,
62
+ 1.3
63
+ ],
64
+ "control_points": [
65
+ 5,
66
+ 10
67
+ ],
68
+ "rotation": [
69
+ -90,
70
+ 90
71
+ ],
72
+ "min_zoom": 1.0,
73
+ "max_zoom": 1.2,
74
+ "shear_range": [
75
+ 0.1,
76
+ 0.1,
77
+ 0.1
78
+ ]
79
+ }
80
+ }
manual_annotated/model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4096cd6fc6b8339cb21c2862420cd8ba0e3242f275e8c26bce63b88684e153bf
3
+ size 23320705
manual_annotated_synthetic/config.json ADDED
@@ -0,0 +1,80 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "batch_size": 32,
3
+ "sliding_window_mode": "constant",
4
+ "models": {
5
+ "lr": "1e-5",
6
+ "wd": 0,
7
+ "dropout": 0.0,
8
+ "model": "unet",
9
+ "unet": {
10
+ "channels": [
11
+ 32,
12
+ 64,
13
+ 128,
14
+ 256
15
+ ],
16
+ "strides": [
17
+ 2,
18
+ 2,
19
+ 2
20
+ ],
21
+ "activation": "LEAKYRELU",
22
+ "normalization": "BATCH"
23
+ }
24
+ },
25
+ "augmentation": {
26
+ "patch_size": [
27
+ 64,
28
+ 64,
29
+ 64
30
+ ],
31
+ "use_augmentation": true,
32
+ "p_gaussian_noise": 0.5,
33
+ "p_gaussian_smooth": 0.5,
34
+ "p_intensity_scale": 0,
35
+ "p_intensity_shift": 0,
36
+ "p_adjust_contrast": 0,
37
+ "p_histogram_shift": 0.5,
38
+ "p_rotate": 0.5,
39
+ "p_zoom": 0.5,
40
+ "p_shear": 0,
41
+ "p_flip": [
42
+ 0,
43
+ 0.5,
44
+ 0.5
45
+ ],
46
+ "gaussian_noise_mean": 0.0,
47
+ "gaussian_noise_std": 0.01,
48
+ "gaussian_smooth_sigma": [
49
+ 0.0,
50
+ 0.5
51
+ ],
52
+ "intensity_scale_factors": [
53
+ -0.5,
54
+ 1.2
55
+ ],
56
+ "intensity_shift_offsets": [
57
+ 0.05,
58
+ 0.2
59
+ ],
60
+ "adjust_contrast_gamma": [
61
+ 0.7,
62
+ 1.3
63
+ ],
64
+ "control_points": [
65
+ 5,
66
+ 10
67
+ ],
68
+ "rotation": [
69
+ -90,
70
+ 90
71
+ ],
72
+ "min_zoom": 1.0,
73
+ "max_zoom": 1.2,
74
+ "shear_range": [
75
+ 0.1,
76
+ 0.1,
77
+ 0.1
78
+ ]
79
+ }
80
+ }
manual_annotated_synthetic/model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9f35baede384cf409a57376df59f6361023ab89ae9c0935a1f2fcae52939fc0a
3
+ size 23320705
splits/test_data.nii ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cf1447c4bf6ba59c9537a7824a7f2559b0c46e93b17d9fef0f0c5fb8114ae78a
3
+ size 6553952
splits/test_label.nii ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d9d410b19a4c1a8aca1a594f60de6bece1866ab22d0c39b520fa7814e23c2250
3
+ size 13107552
splits/train_data.nii ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:45217674b4e57687b6ffd1c7128c5226c445de6dad6aaf97f849420a2f8ace7f
3
+ size 16384352
splits/train_label.nii ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f2308c7aa94fc28370b3d51234630c9b6b8ce2e0f2a257c2ca4c3ff3274949b9
3
+ size 32768352
splits/val_data.nii ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:16f21a32ddc84683d69949c1fbed4fc8f3cdd181100089fcdc23def4881ed341
3
+ size 6553952
splits/val_label.nii ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ce3a2eda66663b0c414a799956f7d10338ad217723cd5aef491444a8088147fc
3
+ size 13107552
synthetic/config.json ADDED
@@ -0,0 +1,80 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "batch_size": 32,
3
+ "sliding_window_mode": "constant",
4
+ "models": {
5
+ "lr": "1e-5",
6
+ "wd": 0,
7
+ "dropout": 0.0,
8
+ "model": "unet",
9
+ "unet": {
10
+ "channels": [
11
+ 32,
12
+ 64,
13
+ 128,
14
+ 256
15
+ ],
16
+ "strides": [
17
+ 2,
18
+ 2,
19
+ 2
20
+ ],
21
+ "activation": "LEAKYRELU",
22
+ "normalization": "BATCH"
23
+ }
24
+ },
25
+ "augmentation": {
26
+ "patch_size": [
27
+ 64,
28
+ 64,
29
+ 64
30
+ ],
31
+ "use_augmentation": true,
32
+ "p_gaussian_noise": 0.5,
33
+ "p_gaussian_smooth": 0.5,
34
+ "p_intensity_scale": 0,
35
+ "p_intensity_shift": 0,
36
+ "p_adjust_contrast": 0,
37
+ "p_histogram_shift": 0.5,
38
+ "p_rotate": 0.5,
39
+ "p_zoom": 0.5,
40
+ "p_shear": 0,
41
+ "p_flip": [
42
+ 0,
43
+ 0.5,
44
+ 0.5
45
+ ],
46
+ "gaussian_noise_mean": 0.0,
47
+ "gaussian_noise_std": 0.01,
48
+ "gaussian_smooth_sigma": [
49
+ 0.0,
50
+ 0.5
51
+ ],
52
+ "intensity_scale_factors": [
53
+ -0.5,
54
+ 1.2
55
+ ],
56
+ "intensity_shift_offsets": [
57
+ 0.05,
58
+ 0.2
59
+ ],
60
+ "adjust_contrast_gamma": [
61
+ 0.7,
62
+ 1.3
63
+ ],
64
+ "control_points": [
65
+ 5,
66
+ 10
67
+ ],
68
+ "rotation": [
69
+ -90,
70
+ 90
71
+ ],
72
+ "min_zoom": 1.0,
73
+ "max_zoom": 1.2,
74
+ "shear_range": [
75
+ 0.1,
76
+ 0.1,
77
+ 0.1
78
+ ]
79
+ }
80
+ }
synthetic/model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:db58216635fd02cf1b741e702a20cc7ee344bf933e52ae3e1ad74dbf0821094d
3
+ size 23320705