├── README.md ├── assets ├── architecture_sbldm_last.svg ├── conditionning.svg └── examples.gif ├── first_stage_training.py ├── modules ├── data_preprocessing.py ├── diffusion.py ├── ema.py ├── loggers.py ├── losses.py ├── models │ ├── attention.py │ ├── autoencoders.py │ ├── base.py │ ├── compvis_unet │ │ ├── nn.py │ │ └── unet.py │ └── unet.py ├── scheduler.py └── utils.py ├── preprocessing_scripts ├── compute_latents_std.py ├── compute_radiomics.py ├── create_first_stage_dataset.py └── create_second_stage_dataset.py ├── sample.py ├── script.sl └── second_stage_training.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arksyd96/synthesis-with-slice-based-ldm/HEAD/README.md -------------------------------------------------------------------------------- /assets/architecture_sbldm_last.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arksyd96/synthesis-with-slice-based-ldm/HEAD/assets/architecture_sbldm_last.svg -------------------------------------------------------------------------------- /assets/conditionning.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arksyd96/synthesis-with-slice-based-ldm/HEAD/assets/conditionning.svg -------------------------------------------------------------------------------- /assets/examples.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arksyd96/synthesis-with-slice-based-ldm/HEAD/assets/examples.gif -------------------------------------------------------------------------------- /first_stage_training.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arksyd96/synthesis-with-slice-based-ldm/HEAD/first_stage_training.py -------------------------------------------------------------------------------- /modules/data_preprocessing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arksyd96/synthesis-with-slice-based-ldm/HEAD/modules/data_preprocessing.py -------------------------------------------------------------------------------- /modules/diffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arksyd96/synthesis-with-slice-based-ldm/HEAD/modules/diffusion.py -------------------------------------------------------------------------------- /modules/ema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arksyd96/synthesis-with-slice-based-ldm/HEAD/modules/ema.py -------------------------------------------------------------------------------- /modules/loggers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arksyd96/synthesis-with-slice-based-ldm/HEAD/modules/loggers.py -------------------------------------------------------------------------------- /modules/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arksyd96/synthesis-with-slice-based-ldm/HEAD/modules/losses.py -------------------------------------------------------------------------------- /modules/models/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arksyd96/synthesis-with-slice-based-ldm/HEAD/modules/models/attention.py -------------------------------------------------------------------------------- /modules/models/autoencoders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arksyd96/synthesis-with-slice-based-ldm/HEAD/modules/models/autoencoders.py -------------------------------------------------------------------------------- /modules/models/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arksyd96/synthesis-with-slice-based-ldm/HEAD/modules/models/base.py -------------------------------------------------------------------------------- /modules/models/compvis_unet/nn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arksyd96/synthesis-with-slice-based-ldm/HEAD/modules/models/compvis_unet/nn.py -------------------------------------------------------------------------------- /modules/models/compvis_unet/unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arksyd96/synthesis-with-slice-based-ldm/HEAD/modules/models/compvis_unet/unet.py -------------------------------------------------------------------------------- /modules/models/unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arksyd96/synthesis-with-slice-based-ldm/HEAD/modules/models/unet.py -------------------------------------------------------------------------------- /modules/scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arksyd96/synthesis-with-slice-based-ldm/HEAD/modules/scheduler.py -------------------------------------------------------------------------------- /modules/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arksyd96/synthesis-with-slice-based-ldm/HEAD/modules/utils.py -------------------------------------------------------------------------------- /preprocessing_scripts/compute_latents_std.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arksyd96/synthesis-with-slice-based-ldm/HEAD/preprocessing_scripts/compute_latents_std.py -------------------------------------------------------------------------------- /preprocessing_scripts/compute_radiomics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arksyd96/synthesis-with-slice-based-ldm/HEAD/preprocessing_scripts/compute_radiomics.py -------------------------------------------------------------------------------- /preprocessing_scripts/create_first_stage_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arksyd96/synthesis-with-slice-based-ldm/HEAD/preprocessing_scripts/create_first_stage_dataset.py -------------------------------------------------------------------------------- /preprocessing_scripts/create_second_stage_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arksyd96/synthesis-with-slice-based-ldm/HEAD/preprocessing_scripts/create_second_stage_dataset.py -------------------------------------------------------------------------------- /sample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arksyd96/synthesis-with-slice-based-ldm/HEAD/sample.py -------------------------------------------------------------------------------- /script.sl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arksyd96/synthesis-with-slice-based-ldm/HEAD/script.sl -------------------------------------------------------------------------------- /second_stage_training.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arksyd96/synthesis-with-slice-based-ldm/HEAD/second_stage_training.py --------------------------------------------------------------------------------