This repository contains a pytorch code for pruning Ai art models.
Two notebooks are provided:
- Pruning_2D_autoencoder.ipynb: a 2D autoencoder for pruning image artworks.
- Pruning_3D_autoencoder.ipynb: a 3D voxel autoencoder for pruning sculptures.
The autoencoder are trained to reconstruct the given artwork. When the train is completed, a gradual pruning of the autoencoder's weights is applied to create minimized and abstract versions of the artwork.