color convergence on custom data about diffusion HOT 5 OPEN

Hi @wileewang , how do you generate the samples? I'm having some troubles with obtaining samples of the reverse process.

from diffusion.

Hello, have you solved it? I am also meeting the same color problems.

from diffusion.

Hi @zxyf1213 , you are probably using the wrong function. If you have issues with the colors of the images, and you notice that the histogram is becoming a gaussian, it mean that you are ADDING noise, so you are using the forward process (q-sample). I do not exactly remember what I used, but try to use self.diffusion.p_sample_loop or a variation of that, since it allows you to apply the reverse process. Good luck.

from diffusion.

@alezuech Thank you very much for receiving your reply. I training the improved_diffusion on cityscapes dataset and I haven't changed the original code of the repository, so there isn't like I used the wrong sample process. I don't know why there is a color problem when training diffusion model on cityscapes dataset, and the problem has been bothering me for a long time, do you have any opinion on it?

from diffusion.

@alezuech Thank you very much for receiving your reply. I training the improved_diffusion on cityscapes dataset and I haven't changed the original code of the repository, so there isn't like I used the wrong sample process. I don't know why there is a color problem when training diffusion model on cityscapes dataset, and the problem has been bothering me for a long time, do you have any opinion on it?

have you resolved the problem? Thanks.

from diffusion.