Bad results on medical images

[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.

[zoubohao]

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.

[mlszy928]

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

[yangfar]

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.

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

[ming-transfer]

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