Hugging Face's logo

RTLucassen
/
slidesegmenter

Model card Files
xet
 Community
RTLucassen commited on
Commit
18f0a19
·
verified ·
1 Parent(s): 6363d87

Upload model parameters and training configs

Browse files
Files changed (4) hide show
  1. 2023-08-13/model_state_dict.pth +3 -0
  2. 2023-08-13/settings.json +156 -0
  3. 2024-01-10/model_state_dict.pth +3 -0
  4. 2024-01-10/settings.json +180 -0
2023-08-13/model_state_dict.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0af045a08d2f8069dd4f14710f32cf303c8d33704a1b577dad055290e063d22e
3
+ size 160276579
2023-08-13/settings.json ADDED
@@ -0,0 +1,156 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "experiment_name": "Modified_U-Net_2023-08-13_19h51m41s",
3
+ "dataset_filename": "dataset.xlsx",
4
+ "seed": 123,
5
+ "model_name": "ModifiedUNet",
6
+ "compile_model": false,
7
+ "checkpoint_path": null,
8
+ "model": {
9
+ "input_channels": 3,
10
+ "filters": 32,
11
+ "downsample_factors": [
12
+ 2,
13
+ 4,
14
+ 4,
15
+ 4,
16
+ 4
17
+ ],
18
+ "residual_connection": false
19
+ },
20
+ "training": {
21
+ "learning_rate": 0.0003,
22
+ "iterations": 100000,
23
+ "iterations_per_decay": 20000,
24
+ "decay_factor": 0.5,
25
+ "iterations_per_checkpoint": 500,
26
+ "iterations_per_update": 5,
27
+ "loss": {
28
+ "weights": [
29
+ 1,
30
+ 1,
31
+ 1,
32
+ 1
33
+ ],
34
+ "class_weights": [
35
+ 1,
36
+ 1
37
+ ],
38
+ "fp_weight": 0.5,
39
+ "fn_weight": 0.5,
40
+ "gamma": 0
41
+ }
42
+ },
43
+ "dataloader": {
44
+ "num_workers": 4,
45
+ "pin_memory": true,
46
+ "batch_size": 1,
47
+ "image_shape": null,
48
+ "max_image_shape": [
49
+ 3072,
50
+ 3072
51
+ ]
52
+ },
53
+ "augmentation": {
54
+ "RandomRotate90": {
55
+ "p": 0.25
56
+ },
57
+ "Affine": {
58
+ "p": 0.5,
59
+ "translate_px": {
60
+ "x": [
61
+ -256,
62
+ 256
63
+ ],
64
+ "y": [
65
+ -256,
66
+ 256
67
+ ]
68
+ },
69
+ "scale": [
70
+ 0.95,
71
+ 1.05
72
+ ],
73
+ "rotate": [
74
+ -180,
75
+ 180
76
+ ]
77
+ },
78
+ "HorizontalFlip": {
79
+ "p": 0.5
80
+ },
81
+ "VerticalFlip": {
82
+ "p": 0.5
83
+ },
84
+ "HueSaturationValue tissue": {
85
+ "p": 0.5,
86
+ "hue_shift_limit": 25,
87
+ "sat_shift_limit": 25,
88
+ "val_shift_limit": 0
89
+ },
90
+ "HueSaturationValue pen+background": {
91
+ "p": 0.5,
92
+ "hue_shift_limit": 100,
93
+ "sat_shift_limit": 0,
94
+ "val_shift_limit": 0
95
+ },
96
+ "HueSaturationValue pen": {
97
+ "p": 0.5,
98
+ "hue_shift_limit": 0,
99
+ "sat_shift_limit": 50,
100
+ "val_shift_limit": 50
101
+ },
102
+ "RandomBrightnessContrast": {
103
+ "p": 0.25,
104
+ "brightness_limit": 0.2,
105
+ "contrast_limit": 0.2
106
+ },
107
+ "RandomGamma": {
108
+ "p": 0.25,
109
+ "gamma_limit": [
110
+ 67,
111
+ 150
112
+ ]
113
+ },
114
+ "GaussNoise": {
115
+ "p": 0.2,
116
+ "var_limit": [
117
+ 0,
118
+ 25
119
+ ]
120
+ },
121
+ "GaussianBlur": {
122
+ "p": 0.2,
123
+ "sigma_limit": [
124
+ 1e-05,
125
+ 2
126
+ ]
127
+ },
128
+ "Padding": [
129
+ {
130
+ "mode": "reflect",
131
+ "p": 0.8
132
+ },
133
+ {
134
+ "mode": "constant",
135
+ "p": 0.1,
136
+ "value": 0
137
+ },
138
+ {
139
+ "mode": "constant",
140
+ "p": 0.1,
141
+ "value": 255
142
+ }
143
+ ]
144
+ },
145
+ "hyperparameters": {
146
+ "tissue_threshold": 0.3,
147
+ "pen_marking_threshold": 0.1,
148
+ "pixels_per_bin": 20,
149
+ "sigma": 4,
150
+ "filter_size": 15,
151
+ "percentile": 95,
152
+ "padding_mode": "constant",
153
+ "padding_value": 0,
154
+ "distance_factor": 100
155
+ }
156
+ }
2024-01-10/model_state_dict.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:437fe15582d121440a0406c9155af924cb24e5fd1378d938b367d830db36ff3f
3
+ size 160276579
2024-01-10/settings.json ADDED
@@ -0,0 +1,180 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "experiment_name": "Modified_U-Net_2024-01-10_13h41m39s",
3
+ "dataset_filename": "dataset.xlsx",
4
+ "seed": 64,
5
+ "model_name": "ModifiedUNet",
6
+ "compile_model": false,
7
+ "checkpoint_path": null,
8
+ "model": {
9
+ "input_channels": 3,
10
+ "filters": 32,
11
+ "downsample_factors": [
12
+ 2,
13
+ 4,
14
+ 4,
15
+ 4,
16
+ 4
17
+ ],
18
+ "residual_connection": false
19
+ },
20
+ "training": {
21
+ "learning_rate": 0.0003,
22
+ "iterations": 100000,
23
+ "iterations_per_decay": 20000,
24
+ "decay_factor": 0.5,
25
+ "iterations_per_checkpoint": 500,
26
+ "iterations_per_update": 5,
27
+ "loss": {
28
+ "weights": [
29
+ 1,
30
+ 10,
31
+ 1,
32
+ 1
33
+ ],
34
+ "class_weights": [
35
+ 1,
36
+ 1
37
+ ],
38
+ "fp_weight": 0.5,
39
+ "fn_weight": 0.5,
40
+ "gamma": 0
41
+ }
42
+ },
43
+ "dataloader": {
44
+ "num_workers": 4,
45
+ "pin_memory": true,
46
+ "batch_size": 1,
47
+ "image_shape": null,
48
+ "max_image_shape": [
49
+ 3072,
50
+ 3072
51
+ ]
52
+ },
53
+ "augmentation": {
54
+ "RandomRotate90": {
55
+ "p": 0.25
56
+ },
57
+ "Affine": {
58
+ "p": 0.5,
59
+ "translate_px": {
60
+ "x": [
61
+ -256,
62
+ 256
63
+ ],
64
+ "y": [
65
+ -256,
66
+ 256
67
+ ]
68
+ },
69
+ "scale": [
70
+ 0.95,
71
+ 1.05
72
+ ],
73
+ "rotate": [
74
+ -180,
75
+ 180
76
+ ]
77
+ },
78
+ "HorizontalFlip": {
79
+ "p": 0.5
80
+ },
81
+ "VerticalFlip": {
82
+ "p": 0.5
83
+ },
84
+ "HueSaturationValue": {
85
+ "p": 0.5,
86
+ "hue_shift_limit": 25,
87
+ "sat_shift_limit": 5,
88
+ "val_shift_limit": 0
89
+ },
90
+ "HueSaturationValue pen": {
91
+ "p": 0.25,
92
+ "hue_shift_limit": 10,
93
+ "sat_shift_limit": 5,
94
+ "val_shift_limit": 0
95
+ },
96
+ "HueSaturationValue tissue": {
97
+ "p": 0.25,
98
+ "hue_shift_limit": 10,
99
+ "sat_shift_limit": 5,
100
+ "val_shift_limit": 0
101
+ },
102
+ "HueSaturationValue non-tissue": {
103
+ "p": 0.0,
104
+ "hue_shift_limit": 0,
105
+ "sat_shift_limit": 0,
106
+ "val_shift_limit": 0
107
+ },
108
+ "HueSaturationValue background": {
109
+ "p": 0.25,
110
+ "hue_shift_limit": 10,
111
+ "sat_shift_limit": 5,
112
+ "val_shift_limit": 0
113
+ },
114
+ "RandomBrightnessContrast": {
115
+ "p": 0.25,
116
+ "brightness_limit": 0.2,
117
+ "contrast_limit": 0.2
118
+ },
119
+ "RandomGamma": {
120
+ "p": 0.25,
121
+ "gamma_limit": [
122
+ 67,
123
+ 150
124
+ ]
125
+ },
126
+ "GaussNoise": {
127
+ "p": 0.2,
128
+ "var_limit": [
129
+ 0,
130
+ 25
131
+ ]
132
+ },
133
+ "GaussianBlur": {
134
+ "p": 0.2,
135
+ "sigma_limit": [
136
+ 1e-05,
137
+ 2
138
+ ]
139
+ },
140
+ "JPEGCompression": {
141
+ "p": 0.2,
142
+ "quality_limit": [
143
+ 25,
144
+ 100
145
+ ]
146
+ },
147
+ "Padding": [
148
+ {
149
+ "mode": "reflect",
150
+ "p": 0.9
151
+ },
152
+ {
153
+ "mode": "constant",
154
+ "p": 0.0,
155
+ "value": 0
156
+ },
157
+ {
158
+ "mode": "constant",
159
+ "p": 0.1,
160
+ "value": 255
161
+ }
162
+ ],
163
+ "PenMarkings": {
164
+ "p": 0.6,
165
+ "N": 10,
166
+ "break_if_not_triggered": true
167
+ }
168
+ },
169
+ "hyperparameters": {
170
+ "tissue_threshold": 0.5,
171
+ "pen_marking_threshold": 0.5,
172
+ "pixels_per_bin": 20,
173
+ "sigma": 2,
174
+ "filter_size": 15,
175
+ "percentile": 98,
176
+ "padding_mode": "constant",
177
+ "padding_value": 1,
178
+ "distance_factor": 100
179
+ }
180
+ }