search

psanch21 / VAE-GMVAE

This repository contains the implementation of the VAE and Gaussian Mixture VAE using TensorFlow and several network architectures
Apache License 2.0
206 stars 32 forks source link

Can this be used for other data sets of image data besides the handwritten data set #10

Closed husuqi closed 3 years ago

psanch21 commented 3 years ago

Hi,

Yes, you just have to define a method in the utils file called load_XXX(). There are already two similar methods implemented: load_MNIST and load_FREY. The only thing is that you have to be careful when selecting the number of filters. If the dimensions are different from 28 pixels you may have to tweak a bit the hyperparameters.

Hope this helps!

husuqi commented 3 years ago

Hi,

Yes, you just have to define a method in the utils file called load_XXX(). There are already two similar methods implemented: load_MNIST and load_FREY. The only thing is that you have to be careful when selecting the number of filters. If the dimensions are different from 28 pixels you may have to tweak a bit the hyperparameters.

Hope this helps!

Thanks,and ,can this be used with multiple tag classification problems? such as an image containing multiple categories