Implement of "Head and Neck Registration"
The required packages are located in requirements.
pip install pytorch==1.8.1 torchvision==0.9.1 torchaudio==0.8.1 cudatoolkit=10.2 -c pytorch
pip install -r requirement.txt
Segrap 2023 Datasets Link : https://segrap2023.grand-challenge.org/
45 OARs segmentation from no-contrast and contrast-enhanced CT scans: Brain, Brainstem, Chiasm, Cochlea left, Cochlea right, Esophagus, Eustachian tube left, Eustachian tube right, Eye left, Eye right, Hippocampus left, Hippocampus right, Internal auditory canal left, Internal auditory canal right, Larynx, Larynx glottic, Larynx supraglottic, Lens left, Lens right, Mandible left, Mandible right, Mastoid left, Mastoid right, Middle Ear left, Middle ear right, Optic nerve left, Optic nerve right, Oral cavity, Parotid left, Parotid right, Pharyngeal constrictor muscle, Pituitary, Spinal cord, Submandibular left, Submandibular right, Temporal lobe left, Temporal lobe right, Thyroid, Temporomandibular joint left, Temporomandibular joint right, Trachea, Tympanic cavity left, Tympanic cavity right, Vestibular semicircular canal left, Vestibular semicircular canal right.
- Before training, pre-processing and initial affine registration must be performed.
- For pre-processing, reference
preprocessing_Segrap.py. - For initial affine registration, reference
Train_Affine.py.
- For pre-processing, reference
python Train.py \
--affine_model experiments/affine \
--dataset_dir Dataset/Segrap_2023 \
--save_validation_img True \
--max_epoch 100 \python Inference.py --affine_model experiments/affine --dataset_dir Dataset/Segrap_2023
