This repo contains the code for my blog post on creating animations of doodles using autoencoders trained on synthetic data.
- pytorch 0.4
- writefile-run
pip install writefile-run - pytorch-utils - clone this repo and follow the instructions
modules.ipynbandautoenc_models.ipynbcontain the code for building the pytorch models.lines_dataset.ipynbcontains the code for making the synthetic dataset.- Run
train.ipynbafter setting a save directory name to train a CoordConv autoencoder with a latent space size of 64. You can change this and try out other models as well. - Play around with a trained model to make gifs using the
analysis.ipynbnotebook.
A sample grid of images for the tree gif is also available as tree.bmp. Use this to create some gifs using the autoencoder you trained.