Hugging Face's logo

braxtonowens
/
torch_localize_motor_3d_weights

Model card Files
xet
Community
braxtonowens commited on
Commit
9022e8b
·
verified ·
1 Parent(s): 06673ca

Upload 4 files

Browse files
Files changed (4) hide show
  1. checkpoint_final.pth +3 -0
  2. dataset.json +0 -0
  3. dataset_fingerprint.json +0 -0
  4. plans.json +373 -0
checkpoint_final.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ea9508a0fd70476497afe3a898f7deb77ed871385903d236c2fe13e466f77cf9
3
+ size 567715573
dataset.json ADDED
The diff for this file is too large to render. See raw diff
 
dataset_fingerprint.json ADDED
The diff for this file is too large to render. See raw diff
 
plans.json ADDED
@@ -0,0 +1,373 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "dataset_name": "Dataset189_Kaggle2025_BYU_FlagellarMotors_mergedExternalBartleyNonBartley_512",
3
+ "plans_name": "nnUNetResEncUNetMPlans",
4
+ "original_median_spacing_after_transp": [
5
+ 1.0,
6
+ 1.0,
7
+ 1.0
8
+ ],
9
+ "original_median_shape_after_transp": [
10
+ 268,
11
+ 512,
12
+ 512
13
+ ],
14
+ "image_reader_writer": "SimpleITKIO",
15
+ "transpose_forward": [
16
+ 0,
17
+ 1,
18
+ 2
19
+ ],
20
+ "transpose_backward": [
21
+ 0,
22
+ 1,
23
+ 2
24
+ ],
25
+ "configurations": {
26
+ "3d_fullres": {
27
+ "data_identifier": "nnUNetPlans_3d_fullres",
28
+ "preprocessor_name": "DefaultPreprocessor",
29
+ "batch_size": 2,
30
+ "patch_size": [
31
+ 80,
32
+ 160,
33
+ 160
34
+ ],
35
+ "median_image_size_in_voxels": [
36
+ 268.0,
37
+ 512.0,
38
+ 512.0
39
+ ],
40
+ "spacing": [
41
+ 1.0,
42
+ 1.0,
43
+ 1.0
44
+ ],
45
+ "normalization_schemes": [
46
+ "ZScoreNormalization"
47
+ ],
48
+ "use_mask_for_norm": [
49
+ false
50
+ ],
51
+ "resampling_fn_data": "resample_data_or_seg_to_shape",
52
+ "resampling_fn_seg": "resample_data_or_seg_to_shape",
53
+ "resampling_fn_data_kwargs": {
54
+ "is_seg": false,
55
+ "order": 3,
56
+ "order_z": 0,
57
+ "force_separate_z": null
58
+ },
59
+ "resampling_fn_seg_kwargs": {
60
+ "is_seg": true,
61
+ "order": 1,
62
+ "order_z": 0,
63
+ "force_separate_z": null
64
+ },
65
+ "resampling_fn_probabilities": "resample_data_or_seg_to_shape",
66
+ "resampling_fn_probabilities_kwargs": {
67
+ "is_seg": false,
68
+ "order": 1,
69
+ "order_z": 0,
70
+ "force_separate_z": null
71
+ },
72
+ "architecture": {
73
+ "network_class_name": "dynamic_network_architectures.architectures.unet.ResidualEncoderUNet",
74
+ "arch_kwargs": {
75
+ "n_stages": 6,
76
+ "features_per_stage": [
77
+ 32,
78
+ 64,
79
+ 128,
80
+ 256,
81
+ 320,
82
+ 320
83
+ ],
84
+ "conv_op": "torch.nn.modules.conv.Conv3d",
85
+ "kernel_sizes": [
86
+ [
87
+ 3,
88
+ 3,
89
+ 3
90
+ ],
91
+ [
92
+ 3,
93
+ 3,
94
+ 3
95
+ ],
96
+ [
97
+ 3,
98
+ 3,
99
+ 3
100
+ ],
101
+ [
102
+ 3,
103
+ 3,
104
+ 3
105
+ ],
106
+ [
107
+ 3,
108
+ 3,
109
+ 3
110
+ ],
111
+ [
112
+ 3,
113
+ 3,
114
+ 3
115
+ ]
116
+ ],
117
+ "strides": [
118
+ [
119
+ 1,
120
+ 1,
121
+ 1
122
+ ],
123
+ [
124
+ 2,
125
+ 2,
126
+ 2
127
+ ],
128
+ [
129
+ 2,
130
+ 2,
131
+ 2
132
+ ],
133
+ [
134
+ 2,
135
+ 2,
136
+ 2
137
+ ],
138
+ [
139
+ 2,
140
+ 2,
141
+ 2
142
+ ],
143
+ [
144
+ 1,
145
+ 2,
146
+ 2
147
+ ]
148
+ ],
149
+ "n_blocks_per_stage": [
150
+ 1,
151
+ 3,
152
+ 4,
153
+ 6,
154
+ 6,
155
+ 6
156
+ ],
157
+ "n_conv_per_stage_decoder": [
158
+ 1,
159
+ 1,
160
+ 1,
161
+ 1,
162
+ 1
163
+ ],
164
+ "conv_bias": true,
165
+ "norm_op": "torch.nn.modules.instancenorm.InstanceNorm3d",
166
+ "norm_op_kwargs": {
167
+ "eps": 1e-05,
168
+ "affine": true
169
+ },
170
+ "dropout_op": null,
171
+ "dropout_op_kwargs": null,
172
+ "nonlin": "torch.nn.LeakyReLU",
173
+ "nonlin_kwargs": {
174
+ "inplace": true
175
+ }
176
+ },
177
+ "_kw_requires_import": [
178
+ "conv_op",
179
+ "norm_op",
180
+ "dropout_op",
181
+ "nonlin"
182
+ ]
183
+ },
184
+ "batch_dice": true
185
+ },
186
+ "3d_fullres_bs16_ps128_256_256": {
187
+ "data_identifier": "nnUNetPlans_3d_fullres",
188
+ "preprocessor_name": "DefaultPreprocessor",
189
+ "batch_size": 16,
190
+ "patch_size": [
191
+ 128,
192
+ 256,
193
+ 256
194
+ ],
195
+ "median_image_size_in_voxels": [
196
+ 200.0,
197
+ 384.0,
198
+ 384.0
199
+ ],
200
+ "spacing": [
201
+ 1.0,
202
+ 1.0,
203
+ 1.0
204
+ ],
205
+ "normalization_schemes": [
206
+ "ZScoreNormalization"
207
+ ],
208
+ "use_mask_for_norm": [
209
+ false
210
+ ],
211
+ "resampling_fn_data": "resample_data_or_seg_to_shape",
212
+ "resampling_fn_seg": "resample_data_or_seg_to_shape",
213
+ "resampling_fn_data_kwargs": {
214
+ "is_seg": false,
215
+ "order": 3,
216
+ "order_z": 0,
217
+ "force_separate_z": null
218
+ },
219
+ "resampling_fn_seg_kwargs": {
220
+ "is_seg": true,
221
+ "order": 1,
222
+ "order_z": 0,
223
+ "force_separate_z": null
224
+ },
225
+ "resampling_fn_probabilities": "resample_data_or_seg_to_shape",
226
+ "resampling_fn_probabilities_kwargs": {
227
+ "is_seg": false,
228
+ "order": 1,
229
+ "order_z": 0,
230
+ "force_separate_z": null
231
+ },
232
+ "architecture": {
233
+ "network_class_name": "dynamic_network_architectures.architectures.unet.ResidualEncoderUNet",
234
+ "arch_kwargs": {
235
+ "n_stages": 7,
236
+ "features_per_stage": [
237
+ 32,
238
+ 64,
239
+ 128,
240
+ 256,
241
+ 320,
242
+ 320,
243
+ 320
244
+ ],
245
+ "conv_op": "torch.nn.modules.conv.Conv3d",
246
+ "kernel_sizes": [
247
+ [
248
+ 3,
249
+ 3,
250
+ 3
251
+ ],
252
+ [
253
+ 3,
254
+ 3,
255
+ 3
256
+ ],
257
+ [
258
+ 3,
259
+ 3,
260
+ 3
261
+ ],
262
+ [
263
+ 3,
264
+ 3,
265
+ 3
266
+ ],
267
+ [
268
+ 3,
269
+ 3,
270
+ 3
271
+ ],
272
+ [
273
+ 3,
274
+ 3,
275
+ 3
276
+ ],
277
+ [
278
+ 3,
279
+ 3,
280
+ 3
281
+ ]
282
+ ],
283
+ "strides": [
284
+ [
285
+ 1,
286
+ 1,
287
+ 1
288
+ ],
289
+ [
290
+ 2,
291
+ 2,
292
+ 2
293
+ ],
294
+ [
295
+ 2,
296
+ 2,
297
+ 2
298
+ ],
299
+ [
300
+ 2,
301
+ 2,
302
+ 2
303
+ ],
304
+ [
305
+ 2,
306
+ 2,
307
+ 2
308
+ ],
309
+ [
310
+ 2,
311
+ 2,
312
+ 2
313
+ ],
314
+ [
315
+ 1,
316
+ 2,
317
+ 2
318
+ ]
319
+ ],
320
+ "n_blocks_per_stage": [
321
+ 1,
322
+ 3,
323
+ 4,
324
+ 6,
325
+ 6,
326
+ 6,
327
+ 6
328
+ ],
329
+ "n_conv_per_stage_decoder": [
330
+ 1,
331
+ 1,
332
+ 1,
333
+ 1,
334
+ 1,
335
+ 1
336
+ ],
337
+ "conv_bias": true,
338
+ "norm_op": "torch.nn.modules.instancenorm.InstanceNorm3d",
339
+ "norm_op_kwargs": {
340
+ "eps": 1e-05,
341
+ "affine": true
342
+ },
343
+ "dropout_op": null,
344
+ "dropout_op_kwargs": null,
345
+ "nonlin": "torch.nn.LeakyReLU",
346
+ "nonlin_kwargs": {
347
+ "inplace": true
348
+ }
349
+ },
350
+ "_kw_requires_import": [
351
+ "conv_op",
352
+ "norm_op",
353
+ "dropout_op",
354
+ "nonlin"
355
+ ]
356
+ },
357
+ "batch_dice": true
358
+ }
359
+ },
360
+ "experiment_planner_used": "nnUNetPlannerResEncM",
361
+ "label_manager": "LabelManager",
362
+ "foreground_intensity_properties_per_channel": {
363
+ "0": {
364
+ "max": 255.0,
365
+ "mean": 110.44560843165799,
366
+ "median": 109.0,
367
+ "min": 0.0,
368
+ "percentile_00_5": 7.0,
369
+ "percentile_99_5": 244.0,
370
+ "std": 43.60688120936182
371
+ }
372
+ }
373
+ }