Hugging Face's logo

aagatti
/
nnunet_knee

PyTorch
nnunetv2
nnunet
medical
knee
mri
segmentation
biomedical
cartilage
meniscus
bone
Eval Results (legacy)
Model card Files
xet
Community
1
aagatti commited on
Commit
6ccfbd6
·
verified ·
1 Parent(s): 88114ca

Add 3d_fullres single-stage model

Browse files
Files changed (7) hide show
  1. models/Dataset500_KneeMRI/nnUNetTrainer__nnUNetResEncUNetMPlans__3d_fullres/dataset.json +19 -0
  2. models/Dataset500_KneeMRI/nnUNetTrainer__nnUNetResEncUNetMPlans__3d_fullres/fold_1/checkpoint_best.pth +3 -0
  3. models/Dataset500_KneeMRI/nnUNetTrainer__nnUNetResEncUNetMPlans__3d_fullres/fold_1/checkpoint_final.pth +3 -0
  4. models/Dataset500_KneeMRI/nnUNetTrainer__nnUNetResEncUNetMPlans__3d_fullres/model_config.json +24 -0
  5. models/Dataset500_KneeMRI/nnUNetTrainer__nnUNetResEncUNetMPlans__3d_fullres/plans.json +532 -0
  6. test_data/test_prediction.nii.gz +2 -2
  7. test_data/test_prediction_cascade.nii.gz +3 -0
models/Dataset500_KneeMRI/nnUNetTrainer__nnUNetResEncUNetMPlans__3d_fullres/dataset.json ADDED
@@ -0,0 +1,19 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "channel_names": {
3
+ "0": "MRI"
4
+ },
5
+ "labels": {
6
+ "background": 0,
7
+ "pat_cart": 1,
8
+ "fem_cart": 2,
9
+ "med_tib_cart": 3,
10
+ "lat_tib_cart": 4,
11
+ "med_men": 5,
12
+ "lat_men": 6,
13
+ "femur": 7,
14
+ "tibia": 8,
15
+ "patella": 9
16
+ },
17
+ "numTraining": 342,
18
+ "file_ending": ".nii.gz"
19
+ }
models/Dataset500_KneeMRI/nnUNetTrainer__nnUNetResEncUNetMPlans__3d_fullres/fold_1/checkpoint_best.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ced27b5ee8cd6edbea9a868e88fafb236d4eac039d07f2996089b39029a65d1a
3
+ size 816053007
models/Dataset500_KneeMRI/nnUNetTrainer__nnUNetResEncUNetMPlans__3d_fullres/fold_1/checkpoint_final.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9d9127b8fbded2f4946a3e539fe2d9099b2872730576131414ee9f941cde32e1
3
+ size 816063205
models/Dataset500_KneeMRI/nnUNetTrainer__nnUNetResEncUNetMPlans__3d_fullres/model_config.json ADDED
@@ -0,0 +1,24 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_info": {
3
+ "fold": 1,
4
+ "stage": "fullres",
5
+ "architecture": "nnUNet ResEnc (Residual Encoder)",
6
+ "configuration": "3d_lowres",
7
+ "planner": "nnUNetResEncUNetMPlans"
8
+ },
9
+ "deployment": {
10
+ "copied_date": "2025-11-07T10:27:06.337179"
11
+ },
12
+ "labels": {
13
+ "0": "background",
14
+ "1": "patellar_cartilage",
15
+ "2": "femoral_cartilage",
16
+ "3": "medial_tibial_cartilage",
17
+ "4": "lateral_tibial_cartilage",
18
+ "5": "medial_meniscus",
19
+ "6": "lateral_meniscus",
20
+ "7": "femur_bone",
21
+ "8": "tibia_bone",
22
+ "9": "patella_bone"
23
+ }
24
+ }
models/Dataset500_KneeMRI/nnUNetTrainer__nnUNetResEncUNetMPlans__3d_fullres/plans.json ADDED
@@ -0,0 +1,532 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "dataset_name": "Dataset500_KneeMRI",
3
+ "plans_name": "nnUNetResEncUNetMPlans",
4
+ "original_median_spacing_after_transp": [
5
+ 0.7995027303695679,
6
+ 0.31251341104507446,
7
+ 0.3125135600566864
8
+ ],
9
+ "original_median_shape_after_transp": [
10
+ 160,
11
+ 512,
12
+ 510
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
+ "2d": {
27
+ "data_identifier": "nnUNetPlans_2d",
28
+ "preprocessor_name": "DefaultPreprocessor",
29
+ "batch_size": 12,
30
+ "patch_size": [
31
+ 512,
32
+ 512
33
+ ],
34
+ "median_image_size_in_voxels": [
35
+ 506.0,
36
+ 503.0
37
+ ],
38
+ "spacing": [
39
+ 0.31251341104507446,
40
+ 0.3125135600566864
41
+ ],
42
+ "normalization_schemes": [
43
+ "ZScoreNormalization"
44
+ ],
45
+ "use_mask_for_norm": [
46
+ false
47
+ ],
48
+ "resampling_fn_data": "resample_data_or_seg_to_shape",
49
+ "resampling_fn_seg": "resample_data_or_seg_to_shape",
50
+ "resampling_fn_data_kwargs": {
51
+ "is_seg": false,
52
+ "order": 3,
53
+ "order_z": 0,
54
+ "force_separate_z": null
55
+ },
56
+ "resampling_fn_seg_kwargs": {
57
+ "is_seg": true,
58
+ "order": 1,
59
+ "order_z": 0,
60
+ "force_separate_z": null
61
+ },
62
+ "resampling_fn_probabilities": "resample_data_or_seg_to_shape",
63
+ "resampling_fn_probabilities_kwargs": {
64
+ "is_seg": false,
65
+ "order": 1,
66
+ "order_z": 0,
67
+ "force_separate_z": null
68
+ },
69
+ "architecture": {
70
+ "network_class_name": "dynamic_network_architectures.architectures.unet.ResidualEncoderUNet",
71
+ "arch_kwargs": {
72
+ "n_stages": 8,
73
+ "features_per_stage": [
74
+ 32,
75
+ 64,
76
+ 128,
77
+ 256,
78
+ 512,
79
+ 512,
80
+ 512,
81
+ 512
82
+ ],
83
+ "conv_op": "torch.nn.modules.conv.Conv2d",
84
+ "kernel_sizes": [
85
+ [
86
+ 3,
87
+ 3
88
+ ],
89
+ [
90
+ 3,
91
+ 3
92
+ ],
93
+ [
94
+ 3,
95
+ 3
96
+ ],
97
+ [
98
+ 3,
99
+ 3
100
+ ],
101
+ [
102
+ 3,
103
+ 3
104
+ ],
105
+ [
106
+ 3,
107
+ 3
108
+ ],
109
+ [
110
+ 3,
111
+ 3
112
+ ],
113
+ [
114
+ 3,
115
+ 3
116
+ ]
117
+ ],
118
+ "strides": [
119
+ [
120
+ 1,
121
+ 1
122
+ ],
123
+ [
124
+ 2,
125
+ 2
126
+ ],
127
+ [
128
+ 2,
129
+ 2
130
+ ],
131
+ [
132
+ 2,
133
+ 2
134
+ ],
135
+ [
136
+ 2,
137
+ 2
138
+ ],
139
+ [
140
+ 2,
141
+ 2
142
+ ],
143
+ [
144
+ 2,
145
+ 2
146
+ ],
147
+ [
148
+ 2,
149
+ 2
150
+ ]
151
+ ],
152
+ "n_blocks_per_stage": [
153
+ 1,
154
+ 3,
155
+ 4,
156
+ 6,
157
+ 6,
158
+ 6,
159
+ 6,
160
+ 6
161
+ ],
162
+ "n_conv_per_stage_decoder": [
163
+ 1,
164
+ 1,
165
+ 1,
166
+ 1,
167
+ 1,
168
+ 1,
169
+ 1
170
+ ],
171
+ "conv_bias": true,
172
+ "norm_op": "torch.nn.modules.instancenorm.InstanceNorm2d",
173
+ "norm_op_kwargs": {
174
+ "eps": 1e-05,
175
+ "affine": true
176
+ },
177
+ "dropout_op": null,
178
+ "dropout_op_kwargs": null,
179
+ "nonlin": "torch.nn.LeakyReLU",
180
+ "nonlin_kwargs": {
181
+ "inplace": true
182
+ }
183
+ },
184
+ "_kw_requires_import": [
185
+ "conv_op",
186
+ "norm_op",
187
+ "dropout_op",
188
+ "nonlin"
189
+ ]
190
+ },
191
+ "batch_dice": true
192
+ },
193
+ "3d_lowres": {
194
+ "data_identifier": "nnUNetResEncUNetMPlans_3d_lowres",
195
+ "preprocessor_name": "DefaultPreprocessor",
196
+ "batch_size": 2,
197
+ "patch_size": [
198
+ 64,
199
+ 192,
200
+ 192
201
+ ],
202
+ "median_image_size_in_voxels": [
203
+ 112,
204
+ 289,
205
+ 287
206
+ ],
207
+ "spacing": [
208
+ 1.0744648146740094,
209
+ 0.5479941579353105,
210
+ 0.547994419228074
211
+ ],
212
+ "normalization_schemes": [
213
+ "ZScoreNormalization"
214
+ ],
215
+ "use_mask_for_norm": [
216
+ false
217
+ ],
218
+ "resampling_fn_data": "resample_data_or_seg_to_shape",
219
+ "resampling_fn_seg": "resample_data_or_seg_to_shape",
220
+ "resampling_fn_data_kwargs": {
221
+ "is_seg": false,
222
+ "order": 3,
223
+ "order_z": 0,
224
+ "force_separate_z": null
225
+ },
226
+ "resampling_fn_seg_kwargs": {
227
+ "is_seg": true,
228
+ "order": 1,
229
+ "order_z": 0,
230
+ "force_separate_z": null
231
+ },
232
+ "resampling_fn_probabilities": "resample_data_or_seg_to_shape",
233
+ "resampling_fn_probabilities_kwargs": {
234
+ "is_seg": false,
235
+ "order": 1,
236
+ "order_z": 0,
237
+ "force_separate_z": null
238
+ },
239
+ "architecture": {
240
+ "network_class_name": "dynamic_network_architectures.architectures.unet.ResidualEncoderUNet",
241
+ "arch_kwargs": {
242
+ "n_stages": 6,
243
+ "features_per_stage": [
244
+ 32,
245
+ 64,
246
+ 128,
247
+ 256,
248
+ 320,
249
+ 320
250
+ ],
251
+ "conv_op": "torch.nn.modules.conv.Conv3d",
252
+ "kernel_sizes": [
253
+ [
254
+ 3,
255
+ 3,
256
+ 3
257
+ ],
258
+ [
259
+ 3,
260
+ 3,
261
+ 3
262
+ ],
263
+ [
264
+ 3,
265
+ 3,
266
+ 3
267
+ ],
268
+ [
269
+ 3,
270
+ 3,
271
+ 3
272
+ ],
273
+ [
274
+ 3,
275
+ 3,
276
+ 3
277
+ ],
278
+ [
279
+ 3,
280
+ 3,
281
+ 3
282
+ ]
283
+ ],
284
+ "strides": [
285
+ [
286
+ 1,
287
+ 1,
288
+ 1
289
+ ],
290
+ [
291
+ 2,
292
+ 2,
293
+ 2
294
+ ],
295
+ [
296
+ 2,
297
+ 2,
298
+ 2
299
+ ],
300
+ [
301
+ 2,
302
+ 2,
303
+ 2
304
+ ],
305
+ [
306
+ 2,
307
+ 2,
308
+ 2
309
+ ],
310
+ [
311
+ 1,
312
+ 2,
313
+ 2
314
+ ]
315
+ ],
316
+ "n_blocks_per_stage": [
317
+ 1,
318
+ 3,
319
+ 4,
320
+ 6,
321
+ 6,
322
+ 6
323
+ ],
324
+ "n_conv_per_stage_decoder": [
325
+ 1,
326
+ 1,
327
+ 1,
328
+ 1,
329
+ 1
330
+ ],
331
+ "conv_bias": true,
332
+ "norm_op": "torch.nn.modules.instancenorm.InstanceNorm3d",
333
+ "norm_op_kwargs": {
334
+ "eps": 1e-05,
335
+ "affine": true
336
+ },
337
+ "dropout_op": null,
338
+ "dropout_op_kwargs": null,
339
+ "nonlin": "torch.nn.LeakyReLU",
340
+ "nonlin_kwargs": {
341
+ "inplace": true
342
+ }
343
+ },
344
+ "_kw_requires_import": [
345
+ "conv_op",
346
+ "norm_op",
347
+ "dropout_op",
348
+ "nonlin"
349
+ ]
350
+ },
351
+ "batch_dice": false,
352
+ "next_stage": "3d_cascade_fullres"
353
+ },
354
+ "3d_fullres": {
355
+ "data_identifier": "nnUNetPlans_3d_fullres",
356
+ "preprocessor_name": "DefaultPreprocessor",
357
+ "batch_size": 2,
358
+ "patch_size": [
359
+ 56,
360
+ 192,
361
+ 160
362
+ ],
363
+ "median_image_size_in_voxels": [
364
+ 150.0,
365
+ 506.0,
366
+ 503.0
367
+ ],
368
+ "spacing": [
369
+ 0.7995027303695679,
370
+ 0.31251341104507446,
371
+ 0.3125135600566864
372
+ ],
373
+ "normalization_schemes": [
374
+ "ZScoreNormalization"
375
+ ],
376
+ "use_mask_for_norm": [
377
+ false
378
+ ],
379
+ "resampling_fn_data": "resample_data_or_seg_to_shape",
380
+ "resampling_fn_seg": "resample_data_or_seg_to_shape",
381
+ "resampling_fn_data_kwargs": {
382
+ "is_seg": false,
383
+ "order": 3,
384
+ "order_z": 0,
385
+ "force_separate_z": null
386
+ },
387
+ "resampling_fn_seg_kwargs": {
388
+ "is_seg": true,
389
+ "order": 1,
390
+ "order_z": 0,
391
+ "force_separate_z": null
392
+ },
393
+ "resampling_fn_probabilities": "resample_data_or_seg_to_shape",
394
+ "resampling_fn_probabilities_kwargs": {
395
+ "is_seg": false,
396
+ "order": 1,
397
+ "order_z": 0,
398
+ "force_separate_z": null
399
+ },
400
+ "architecture": {
401
+ "network_class_name": "dynamic_network_architectures.architectures.unet.ResidualEncoderUNet",
402
+ "arch_kwargs": {
403
+ "n_stages": 6,
404
+ "features_per_stage": [
405
+ 32,
406
+ 64,
407
+ 128,
408
+ 256,
409
+ 320,
410
+ 320
411
+ ],
412
+ "conv_op": "torch.nn.modules.conv.Conv3d",
413
+ "kernel_sizes": [
414
+ [
415
+ 1,
416
+ 3,
417
+ 3
418
+ ],
419
+ [
420
+ 3,
421
+ 3,
422
+ 3
423
+ ],
424
+ [
425
+ 3,
426
+ 3,
427
+ 3
428
+ ],
429
+ [
430
+ 3,
431
+ 3,
432
+ 3
433
+ ],
434
+ [
435
+ 3,
436
+ 3,
437
+ 3
438
+ ],
439
+ [
440
+ 3,
441
+ 3,
442
+ 3
443
+ ]
444
+ ],
445
+ "strides": [
446
+ [
447
+ 1,
448
+ 1,
449
+ 1
450
+ ],
451
+ [
452
+ 1,
453
+ 2,
454
+ 2
455
+ ],
456
+ [
457
+ 2,
458
+ 2,
459
+ 2
460
+ ],
461
+ [
462
+ 2,
463
+ 2,
464
+ 2
465
+ ],
466
+ [
467
+ 2,
468
+ 2,
469
+ 2
470
+ ],
471
+ [
472
+ 1,
473
+ 2,
474
+ 2
475
+ ]
476
+ ],
477
+ "n_blocks_per_stage": [
478
+ 1,
479
+ 3,
480
+ 4,
481
+ 6,
482
+ 6,
483
+ 6
484
+ ],
485
+ "n_conv_per_stage_decoder": [
486
+ 1,
487
+ 1,
488
+ 1,
489
+ 1,
490
+ 1
491
+ ],
492
+ "conv_bias": true,
493
+ "norm_op": "torch.nn.modules.instancenorm.InstanceNorm3d",
494
+ "norm_op_kwargs": {
495
+ "eps": 1e-05,
496
+ "affine": true
497
+ },
498
+ "dropout_op": null,
499
+ "dropout_op_kwargs": null,
500
+ "nonlin": "torch.nn.LeakyReLU",
501
+ "nonlin_kwargs": {
502
+ "inplace": true
503
+ }
504
+ },
505
+ "_kw_requires_import": [
506
+ "conv_op",
507
+ "norm_op",
508
+ "dropout_op",
509
+ "nonlin"
510
+ ]
511
+ },
512
+ "batch_dice": true
513
+ },
514
+ "3d_cascade_fullres": {
515
+ "inherits_from": "3d_fullres",
516
+ "previous_stage": "3d_lowres"
517
+ }
518
+ },
519
+ "experiment_planner_used": "nnUNetPlannerResEncM",
520
+ "label_manager": "LabelManager",
521
+ "foreground_intensity_properties_per_channel": {
522
+ "0": {
523
+ "max": 18131.349609375,
524
+ "mean": 288.2044372558594,
525
+ "median": 179.42546844482422,
526
+ "min": 0.0,
527
+ "percentile_00_5": 13.0,
528
+ "percentile_99_5": 2880.8670751953177,
529
+ "std": 466.28704833984375
530
+ }
531
+ }
532
+ }
test_data/test_prediction.nii.gz CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:2841586e4d2c1faae09826c30823c28849f02ee10296ec1144414b6e3d8ac753
3
- size 175391
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:838816961375d5b01710fceb0b3d31a29dd6fc95d68d9bdbde04b7f70d3c181c
3
+ size 176050
test_data/test_prediction_cascade.nii.gz ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2841586e4d2c1faae09826c30823c28849f02ee10296ec1144414b6e3d8ac753
3
+ size 175391