Spatial-aware Attention Generative Adversarial Network for Semi-supervised Anomaly Detection in Medical Image
Zerui Zhang, Zhichao Sun, Zelong Liu, Bo Du, Rui Yu, Zhou Zhao, Yongchao Xu
- Clone this repository and navigate to SAGAN folder
git clone https://github.com/zzr728/SAGAN.git
cd SAGAN- Install Package: Create conda environment
conda create -n SAGAN python=3.9 -y
conda activate SAGAN
pip install --upgrade pip # enable PEP 660 support- Install additional packages for training cases
pip install -r requirements.txtDownload the well-processed Med-AD benchmark from: Google Drive | OneDrive.
(RSNA, VinDr-CXR and LAG are one of the benchmarks, and should be only applied for academic research.)
bash train.shbash test.shThis repository provides pre-trained SAGAN model checkpoints that can be downloaded and used for medical anomaly detection tasks.The available pre-trained models include:
| Model Descriptions | Model Weights |
|---|---|
| SAGAN RSNA-finetuned | SAGAN_G_RSNA.ckpt |
| SAGAN VinDr-CXR-finetuned | SAGAN_G_VinDr-CXR.ckpt |
| SAGAN LAG-finetuned | SAGAN_G_LAG.ckpt |

