This is an implementation of Self-Organising Textures by Eyvind Niklasson et. al. using a neural network framework called tinygrad.
Install dependencies:
python3 -m pip install Pillow numpy matplotlib tinygrad torchvision
To train a model:
python3 train.py
To create a gif called agif.gif of the trained or currently training
model:
python3 -i
>>> from viz import create_gif
>>> create_gif("agif.gif")
Model hyperparamaters can be adjusted in train.py, vgg16.py
contains a tinygrad VGG-16 model which loads its weights from the
torchvision pretrained model. loss.py contains various loss
functions for minimizing style-loss, or maximizing the activation of a
unit the Resnet image classifier.
Tinygrad is much slower than pytorch for training. See pytorch.py
for a pytorch version of this software which has been modified to
produce monochrome textures.