DenoisingDiffusionProbabilityModel-ddpm- icon indicating copy to clipboard operation
DenoisingDiffusionProbabilityModel-ddpm- copied to clipboard

Bad results on medical images

Open mlszy928 opened this issue 1 year ago • 4 comments

I run the default code and achieved satisfactory results on Cifar10, but when I train the model on 2d CT images, the generated images are almost white or black by the model after 50 epoches. image

mlszy928 avatar Jun 06 '23 07:06 mlszy928

Hi

I think this issue is because your data is 2d CT images. In CIFAR-10 dataset, the channel of these images is 3. You can 1. convert the 2d CT images to 3 channel or 2. change the code to fit your task.

Notice: This code is suitable for learning the processing of DDPM but may not be a good repo to do some engineering tasks.

Good luck.

zoubohao avatar Jun 06 '23 08:06 zoubohao

Thank you for reply I have already converted the CT images to 3 channels... Maybe I should change the code for medical images.

mlszy928 avatar Jun 06 '23 08:06 mlszy928

I run the default code and achieved satisfactory results on Cifar10, but when I train the model on 2d CT images, the generated images are almost white or black by the model after 50 epoches. image

Excuse me, have you found some DDPMs suitable for 1D or 2D data?

yangfar avatar Aug 10 '23 12:08 yangfar

Excuse me, I also want to know that DDPM is suitable for 1D signal ?is there some bro can tell me ?

ming-transfer avatar Dec 15 '23 07:12 ming-transfer