snowPu / master-thesis-autoencoder

Master Thesis Autoencoder for downscaling

Repository from Github https://github.comsnowPu/master-thesis-autoencoderRepository from Github https://github.comsnowPu/master-thesis-autoencoder

Autoencoder for image downscaling and Mipmap Generation

This is an implementation of an autoencoder to downscale images and generate mipmaps.

The architecture of the autoencoder can be seen in this diagram:

alt text

To run pretrained model:

The saved weights are in weights/weights_perceptual_ssim_nadam_500_0.0002_0.9_0.999_1587746299.670421

Run the test file image_downscaling/conv_autoencoder_test.py after setting the right folder where you have images.

To train the model:

Download the train dataset in the root folder of the repository from here: https://drive.google.com/file/d/1bZMm92vCFVYYxJW0G06CtSX0IXhlJxEg/view?usp=sharing

Run image_downscaling/conv_autoencoder_main.py after tweaking the parameters as you wish.

About

Master Thesis Autoencoder for downscaling


Languages

Language:Python 100.0%