Skip to content

A deep learning model for histological image compression and decompression

License

Notifications You must be signed in to change notification settings

TheJacksonLaboratory/cnn_autoencoder

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

cnn_autoencoder

Convolutional neural network Autoencoder. Implementation based on Cheng et al. Energy Compaction-Based Image Compression Using Convolutional AutoEncoder, Transactions on Multimedia 14 (8), 2019.

Installation

Clone this repository and use the pytorch_lightnening container. The extra required packages can be installed using pip from requirements.txt.

Training

Training has been tested on MNIST, ImageNet, and a local histology dataset.

Configuration files

There are a set of configuration files that can be used to set up the training parameters. Those can be used by passing the argumen -c config.json, where config.json is a json file containing the parameters of the experiment. The training paramaters can be reviewed by using the following command.

python ./src/train_cae.py -h

Testing

The trained model can be tested using the modules src/compress.py and src/decompress.py by separate.

Compressing and decompressing

Compression and decompression requires a pre-trained model. Other arguments required to run these modules can be listed with the fllowing commands.

python ./src/compress.py -h
python ./src/decompress.py -h

About

A deep learning model for histological image compression and decompression

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published