PyTorch lightning implementation of A Patch-based Student-Teacher Pyramid Matching Approach to Anomaly Detection in 3D Magnetic Resonance Imaging (MIDL 2024).
Install packages with:
$ pip install -r requirements.txt
The data can be accessed at the following link.
The path to the data is ./data/SKULL/{BRATS, IXI}.
Start training the teacher with the following line:
python run_stpm3D --run_mode pretrain
Start training the student with the following line:
python run_stpm3D --run_mode train
The evaluation of the network can be started with the following line:
python run_stpm3D --run_mode eval
- Student-Teacher Feature Pyramid Matching for Anomaly Detection
- Reinventing 2D Convolutions for 3D Images
- ...