2D Image Autoencoder Model

During my research, I needed to learn how an autoencoder model works and how to effectively train one. I used a publicly available dataset of paintings to train the model and observe its progress during training. I experimented with starting from pretrained ResNet models, but I found that a custom model trained from scratch perfomed better for this task.

Python PyTorch Data ML Training
CV Test

CV Test

Figure 1: Model Architecture.

This model consists of