search

w86763777 / pytorch-ddpm

Unofficial PyTorch implementation of Denoising Diffusion Probabilistic Models
Do What The F*ck You Want To Public License
477 stars 60 forks source link

Question about the image size #12

Open DongyangHuLi opened 1 year ago

DongyangHuLi commented 1 year ago

Thanks for sharing the code! I'm confused about the size of the image input from the diffusion model. In the code why limit the height and width of the image must be equal, can not equal? Looking forward to your reply!

w86763777 commented 1 year ago

The height and width of the image do not need to be equal, and this can be accomplished by updating the UNet model.

Another thing to consider is the downsampling and upsampling of UNet, which may cause the input size to differ from the output size if the dimensions are not divisible by 2.