Train VAE for image generation

1 次查看(过去 30 天)
debojit sharma
debojit sharma 2023-6-16
回答: 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:
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.

回答(1 个)

Ranjeet
Ranjeet 2023-6-26
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 elboLossfunction in the referred example – Train Variational Autoencoder (VAE) to Generate Images

类别

Help CenterFile Exchange 中查找有关 Image Data Workflows 的更多信息

Community Treasure Hunt

Find the treasures in MATLAB Central and discover how the community can help you!

Start Hunting!

Translated by