using vq-vae-2 for scoring probability of an image

[zoharbarzelay]

trafficstars

Thanks a lot for this repo! quick question: once i train on your image set - can i somehow then use the model to estimate the probability that a new image is from the same distribution of the training set?

basically i want to use VAE to learn a prior model over images..

Thanks for anyone who can help with this!

Best, Z.

[venergiac]

I reconstruct the image and I compare reconstructed with original using SSIM

[zoharbarzelay]

I reconstruct the image and I compare reconstructed with original using SSIM

Thanks for your answer! could you plz provide some more details? for instance, what are the different colors in the graph?

[rosinality]

I think you can use prior (like PixelSNAIL) and reconstruction loss of the decoder to calculate likelihood.

[zoharbarzelay]

I think you can use prior (like PixelSNAIL) and reconstruction loss of the decoder to calculate likelihood.

thanks for your answer! how would you suggest that i use PixelSNAIL?

[rosinality]

@zozobozo I think you can use probability estimates of PixelSNAIL that you can calculate by forward the code of the sample.