DTM

DTM: Diffusion Transformer Model Guided by Compact Prior in Low-dose PET Reconstruction The Code is created based on the method described in the following paper:
Diffusion Transformer Model Guided by Compact Prior in Low-dose PET Reconstruction
B. Huang， X. Liu， L. Fang， Q. Liu，B. Li
IEEE Journal of Biomedical and Health Informatics

Optional parameters:

weight: Weight for forward loss.
epoch: Specifies number of iterations.

The training pipeline of DTM

Two visualization pipeline of DTM

The results of PET images

Training

1. To pretrain DTM_S1, run

sh trainS1.sh

4. To train DTM_S2, run

#set the 'pretrain_network_g' and 'pretrain_network_S1' in ./options/train_DTMS2.yml to be the path of DTM_S1's pre-trained model

sh trainS2.sh

Note: The above training script uses 8 GPUs by default.

Testing

• Testing

# modify the dataset path in ./options/test_DTMS2.yml

sh test.sh