This repository provides a DiT-based NIFECG anomaly detection code developed as part of the ADVANCED GENERATIVE MODELS (STA8610) course project at Yonsei University Graduate School.
- Anomaly NIFECG dataset and generation script
- 1D DiT-based base diffusion model
- DDAD-based reconstruction module
Clone this repository: git clone https://github.com/Minjee-Seo/AnomalyDiT.git
- Edit the
config.yamlfile in ddad_utils. - Run
python train.pyfor training. - Run
python test.pyfor sampling and evaluation.
For any queries, please reach out to Minjee Seo and Seohyeon Jeong.