Skip to content

This project focuses on image compression using a convolutional autoencoder. The goal is to compress satellite images from the EuroSAT dataset into a compact representation and reconstruct them with minimal quality loss.

License

Notifications You must be signed in to change notification settings

AngelLagr/Convolutional-Autoencoder-for-Image-Compression

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

7 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Convolutional-Autoencoder-for-Image-Compression

This project focuses on image compression using a convolutional autoencoder. The goal is to compress satellite images from the EuroSAT dataset into a compact representation and reconstruct them with minimal quality loss.

The convolutional autoencoder model is trained to learn efficient image representations. It uses a mean squared error (MSE) loss function to minimize the difference between the original and the reconstructed images. The model is implemented using PyTorch and trained using the Adam optimizer.

The model is trained on EuroSAT images, resized to 64x64 pixels during the training process, and evaluated on the ability to reconstruct the images after compression.


Project Structure

  • train.py: Script for training the convolutional autoencoder on the EuroSAT dataset.
  • test.py: Script to test the model on images from the dataset.
  • convAutoencoder.py: Defines the architecture of the convolutional autoencoder model.
  • autoencoder.pth: File storing the trained model weights.
  • bigimage.png: A bigger picture to test the compression in test.py

Technologies Used

  • Python
  • PyTorch (for building and training the convolutional autoencoder)
  • EuroSAT dataset (satellite imagery)
  • Matplotlib (for result visualization)

Execution

To train the model:

python train.py

To test the model on images:

python test.py

About

This project focuses on image compression using a convolutional autoencoder. The goal is to compress satellite images from the EuroSAT dataset into a compact representation and reconstruct them with minimal quality loss.

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages