KS987 commited on
Commit
73c21d5
Β·
verified Β·
1 Parent(s): 0cca16d

Upload README.md

Browse files
Files changed (1) hide show
  1. README.md +56 -0
README.md ADDED
@@ -0,0 +1,56 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ tags:
4
+ - medical-imaging
5
+ - segmentation
6
+ - nnunet
7
+ - ct
8
+ datasets:
9
+ - FLARE-MedFM/FLARE-Task1-Pancancer
10
+ ---
11
+
12
+ # PanCancerSeg Specialized Weights
13
+
14
+ Cancer-specific nnUNet v2 segmentation models trained on CT images from the [CVPR 2026 FLARE Task 1: Pan-cancer Segmentation](https://www.codabench.org/competitions/7149/) dataset.
15
+
16
+ ## Models
17
+
18
+ | Folder | Cancer type |
19
+ |--------|-------------|
20
+ | Dataset102_Kidney | Kidney cancer |
21
+ | Dataset103_Liver | Liver cancer |
22
+ | Dataset104_Pancreas | Pancreatic cancer |
23
+ | Dataset105_Lung | Lung cancer |
24
+
25
+ All models use `nnUNetTrainerWandb2000`, `nnUNetResEncUNetMPlans`, `3d_fullres`, fold 0, trained for 2000 epochs. Each folder contains `checkpoint_best.pth` (best validation checkpoint).
26
+
27
+ ## Usage
28
+
29
+ Download the weights and point `--model_dir` at the root directory:
30
+
31
+ ```bash
32
+ # Clone this repo
33
+ git lfs install
34
+ git clone https://huggingface.co/KS987/PanCancerSeg-Specialized-weights
35
+
36
+ # Run inference
37
+ python predict.py \
38
+ --input /path/to/case.nii.gz \
39
+ --cancer_type kidney_cancer \
40
+ --model_dir ./PanCancerSeg-Specialized-weights \
41
+ --device cuda
42
+ ```
43
+
44
+ See [PanCancerSeg-Inference](https://github.com/Kappapapa123/PanCancerSeg-Inference) for the full inference pipeline.
45
+
46
+ ## File Structure
47
+
48
+ ```
49
+ Dataset10X_*/
50
+ └── nnUNetTrainerWandb2000__nnUNetResEncUNetMPlans__3d_fullres/
51
+ β”œβ”€β”€ dataset.json
52
+ β”œβ”€β”€ dataset_fingerprint.json
53
+ β”œβ”€β”€ plans.json
54
+ └── fold_0/
55
+ └── checkpoint_best.pth
56
+ ```