RTLucassen commited on
Commit
c093a87
·
verified ·
1 Parent(s): b61bb68

Deleted model

Browse files
2025-04-15/model_state_dict.pth DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:c6dbb32377bcb4b2c989c825dbff5789688f293cc97c6553cffb0cb859730492
3
- size 160284006
 
 
 
 
2025-04-15/settings.json DELETED
@@ -1,187 +0,0 @@
1
- {
2
- "experiment_name": "Modified_U-Net_no_test_2025-04-15_10h49m20s",
3
- "dataset_filename": "dataset+addition_no_test.xlsx",
4
- "seed": 64,
5
- "model_name": "ModifiedUNet",
6
- "compile_model": false,
7
- "checkpoint_path": false,
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
- "attach_tissue_decoder": true,
20
- "attach_pen_decoder": true,
21
- "attach_distance_decoder": true
22
- },
23
- "training": {
24
- "learning_rate": 0.0003,
25
- "iterations": 100000,
26
- "iterations_per_decay": 20000,
27
- "decay_factor": 0.5,
28
- "iterations_per_checkpoint": 500,
29
- "iterations_per_update": 5,
30
- "loss": {
31
- "loss_weights": {
32
- "tversky": 1.0,
33
- "focal": 10.0,
34
- "MSE dist": 1.0,
35
- "MSE grad dist": 1.0
36
- },
37
- "class_weights": {
38
- "tissue": 1.0,
39
- "pen": 1.0
40
- },
41
- "fp_weight": 0.5,
42
- "fn_weight": 0.5,
43
- "gamma": 0,
44
- "smooth_nom": 1,
45
- "smooth_denom": 1
46
- }
47
- },
48
- "dataloader": {
49
- "num_workers": 4,
50
- "pin_memory": true,
51
- "batch_size": 1,
52
- "image_shape": null,
53
- "max_image_shape": [
54
- 3072,
55
- 3072
56
- ]
57
- },
58
- "augmentation": {
59
- "RandomRotate90": {
60
- "p": 0.25
61
- },
62
- "Affine": {
63
- "p": 0.5,
64
- "translate_px": {
65
- "x": [
66
- -256,
67
- 256
68
- ],
69
- "y": [
70
- -256,
71
- 256
72
- ]
73
- },
74
- "scale": [
75
- 0.95,
76
- 1.05
77
- ],
78
- "rotate": [
79
- -180,
80
- 180
81
- ]
82
- },
83
- "HorizontalFlip": {
84
- "p": 0.5
85
- },
86
- "VerticalFlip": {
87
- "p": 0.5
88
- },
89
- "HueSaturationValue": {
90
- "p": 0.5,
91
- "hue_shift_limit": 25,
92
- "sat_shift_limit": 5,
93
- "val_shift_limit": 0
94
- },
95
- "HueSaturationValue pen": {
96
- "p": 0.25,
97
- "hue_shift_limit": 10,
98
- "sat_shift_limit": 5,
99
- "val_shift_limit": 0
100
- },
101
- "HueSaturationValue tissue": {
102
- "p": 0.25,
103
- "hue_shift_limit": 10,
104
- "sat_shift_limit": 5,
105
- "val_shift_limit": 0
106
- },
107
- "HueSaturationValue non-tissue": {
108
- "p": 0.0,
109
- "hue_shift_limit": 0,
110
- "sat_shift_limit": 0,
111
- "val_shift_limit": 0
112
- },
113
- "HueSaturationValue background": {
114
- "p": 0.25,
115
- "hue_shift_limit": 10,
116
- "sat_shift_limit": 5,
117
- "val_shift_limit": 0
118
- },
119
- "RandomBrightnessContrast": {
120
- "p": 0.25,
121
- "brightness_limit": 0.2,
122
- "contrast_limit": 0.2
123
- },
124
- "RandomGamma": {
125
- "p": 0.25,
126
- "gamma_limit": [
127
- 67,
128
- 150
129
- ]
130
- },
131
- "GaussNoise": {
132
- "p": 0.2,
133
- "var_limit": [
134
- 0,
135
- 25
136
- ]
137
- },
138
- "GaussianBlur": {
139
- "p": 0.2,
140
- "sigma_limit": [
141
- 1e-05,
142
- 2
143
- ]
144
- },
145
- "JPEGCompression": {
146
- "p": 0.2,
147
- "quality_limit": [
148
- 25,
149
- 100
150
- ]
151
- },
152
- "Padding": [
153
- {
154
- "mode": "reflect",
155
- "p": 0.9
156
- },
157
- {
158
- "mode": "constant",
159
- "p": 0.0,
160
- "value": 0
161
- },
162
- {
163
- "mode": "constant",
164
- "p": 0.1,
165
- "value": 255
166
- }
167
- ],
168
- "PenMarkings": {
169
- "p": 0.1,
170
- "N": [
171
- 2,
172
- 6
173
- ]
174
- }
175
- },
176
- "hyperparameters": {
177
- "tissue_threshold": 0.25,
178
- "pen_marking_threshold": 0.80,
179
- "pixels_per_bin": 20,
180
- "sigma": 3,
181
- "filter_size": 15,
182
- "percentile": 98,
183
- "padding_mode": "constant",
184
- "padding_value": 0,
185
- "distance_factor": 100
186
- }
187
- }