Train VAE for image generation

Question

debojit sharma 2023-6-16

0
链接

此问题的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/1984379-train-vae-for-image-generation

回答： Ranjeet ，2023-6-26

I am trying to implement the code given in the following link with my own dataset containing RGB images of size 200*200:

https://in.mathworks.com/help/deeplearning/ug/train-a-variational-autoencoder-vae-to-generate-images.html

I am getting the following errors in the Train model part:

I can make out that there is some issue happening with the custom functions ('modelLoss', 'elboLoss') given in this link. But I am unable to understand how can I resolve these issues. Someone please kindly guide me . I will be thankful to you.

0 个评论
显示 -1更早的评论隐藏 -1更早的评论

请先登录，再进行评论。

请先登录，再回答此问题。

Answer 1

Ranjeet 2023-6-26

0
链接

此回答的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/1984379-train-vae-for-image-generation#answer_1262503

Hi Debojit,

As per the error message provided, the decoder (netD) output size is not matching with what is being fed to the encoder (netE) input.

I suggest analysing the networks (can refer to this resource - analyzeNetwork) and check if the decoder output is same to that of encoder input (image of size 200*200).

Also, size of ‘Y’ and ‘T’ can be cross-checked to be equal in ‘elboLoss’ function in the referred example – Train Variational Autoencoder (VAE) to Generate Images