Skip to content

snowPu/master-thesis-autoencoder

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

86 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages