hanjang commited on
Commit
b2fdf3c
Β·
verified Β·
1 Parent(s): 17f5254

Update README.md

Browse files

Add model card with usage instructions

Files changed (1) hide show
  1. README.md +92 -3
README.md CHANGED
@@ -1,3 +1,92 @@
1
- ---
2
- license: apache-2.0
3
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: pytorch
3
+ license: apache-2.0
4
+ tags:
5
+ - medical-image-segmentation
6
+ - interactive-segmentation
7
+ - nnunet
8
+ - meningioma
9
+ - radiotherapy
10
+ pipeline_tag: image-segmentation
11
+ ---
12
+
13
+ # Interactive-MEN-RT
14
+
15
+ **Domain-Specialized Interactive Segmentation for Meningioma Radiotherapy Planning**
16
+
17
+ **Status**: Research prototype β€” Not for clinical use.
18
+
19
+ ## Overview
20
+
21
+ Interactive 3D meningioma segmentation framework built on nnU-Net v2 and nnInteractive.
22
+
23
+ - **Performance**: 77.6% Dice, 64.8% IoU (BraTS 2025 Meningioma RT)
24
+ - **Interaction Modes**: Point, scribble, box, lasso
25
+ - **Input**: T1c MRI (contrast-enhanced)
26
+
27
+ ## Files
28
+ ```
29
+ nnUNetInteractionTrainer__nnUNetPlans__3d_fullres_scratch/
30
+ β”œβ”€β”€ plans.json
31
+ β”œβ”€β”€ dataset.json
32
+ └── fold_0/
33
+ β”œβ”€β”€ checkpoint_best.pth (820 MB)
34
+ └── checkpoint_final.pth (820 MB)
35
+ ```
36
+
37
+ ## Quick Start
38
+ ```python
39
+ from huggingface_hub import snapshot_download
40
+ import os
41
+
42
+ # Download checkpoint
43
+ root = snapshot_download(
44
+ "hanjang/Interactive-MEN-RT",
45
+ allow_patterns=["nnUNetInteractionTrainer__nnUNetPlans__3d_fullres_scratch/**"]
46
+ )
47
+ CKPT = os.path.join(root, "nnUNetInteractionTrainer__nnUNetPlans__3d_fullres_scratch")
48
+
49
+ # Load and run inference
50
+ from Interactive_MEN_RT_predictor import InteractiveMENRTPredictor
51
+ import torch, numpy as np
52
+
53
+ predictor = InteractiveMENRTPredictor(
54
+ device=torch.device("cuda" if torch.cuda.is_available() else "cpu")
55
+ )
56
+ predictor.initialize_from_trained_model_folder(
57
+ model_training_output_dir=CKPT,
58
+ use_fold=0,
59
+ checkpoint_name="checkpoint_best.pth"
60
+ )
61
+
62
+ # Run on your volume (shape: 1, H, W, D)
63
+ predictor.reset_interactions()
64
+ predictor.set_image(volume)
65
+ predictor.set_target_buffer(np.zeros_like(volume[0], np.float32))
66
+ predictor._finish_preprocessing_and_initialize_interactions()
67
+ predictor._predict_without_interaction()
68
+ prediction = (predictor.target_buffer > 0.5).astype(np.uint8)
69
+ ```
70
+
71
+ # Citation
72
+
73
+ ```bibtex
74
+ @inproceedings{interactive-men-rt-2025,
75
+ title={Domain-Specialized Interactive Segmentation Framework for Meningioma Radiotherapy Planning},
76
+ author={Lee, Junhyeok and Jang, Han and Choi, Kyu Sung},
77
+ booktitle={MICCAI CLIP Workshop},
78
+ year={2025}
79
+ }
80
+ ```
81
+
82
+ # Links
83
+
84
+ GitHub: [snuh-rad-aicon/Interactive-MEN-RT](https://github.com/snuh-rad-aicon/Interactive-MEN-RT)
85
+
86
+ Contact: janghan001112@gmail.com
87
+
88
+ Developed at Seoul National University AICON Lab
89
+
90
+ Research only. Not for clinical use.
91
+
92
+ ---