# PET-diffusion

**Repository Path**: Heconnor/PET-diffusion

## Basic Information

- **Project Name**: PET-diffusion
- **Description**: No description available
- **Primary Language**: Unknown
- **License**: Not specified
- **Default Branch**: main
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 0
- **Forks**: 0
- **Created**: 2024-11-20
- **Last Updated**: 2025-01-13

## Categories & Tags

**Categories**: Uncategorized

**Tags**: ImageRestoration, DiffusionModel

## README


## PET-diffusion: Unsupervised PET Enhancement based on the Latent Diffusion Model (see https://link.springer.com/chapter/10.1007/978-3-031-43907-0_1)
This repo contains the supported pytorch code and configuration files of our work




# Model Training
The entire model training is divided into two stages. In the first stage, the `train_autoencoder.py` script is used to train an autoencoder for compressing PET images. In the second stage, the `train_LDM.py` script is used to train the LDM model. At this stage, the autoencoder trained in the first stage is called to compress PET images. 

It's important to note that if CT image-guided denoising is required in the second stage, a separate autoencoder (`autoencoder_CT`) needs to be pre-trained specifically for compressing CT images.



## Citation
```
@inproceedings{jiang2023pet,
  title={PET-Diffusion: Unsupervised PET Enhancement Based on the Latent Diffusion Model},
  author={Jiang, Caiwen and Pan, Yongsheng and Liu, Mianxin and Ma, Lei and Zhang, Xiao and Liu, Jiameng and Xiong, Xiaosong and Shen, Dinggang},
  booktitle={International Conference on Medical Image Computing and Computer-Assisted Intervention},
  pages={3--12},
  year={2023},
  organization={Springer}
}
```