baler-collaboration / baler

Repository of Baler, a machine learning based data compression tool
https://github.com/baler-collaboration/baler.github.io
Apache License 2.0
30 stars 26 forks source link

350 add pj model to baler #357

Closed gallenaxel closed 6 months ago

gallenaxel commented 6 months ago

PJ's model was altered and added to models.py. This is an implementation of TensorFlows data compression notebook model, and it has been tested on the MNIST dataset. Below is an example figure of the results from online compression, where training has been done on 59980 images and compression/decompression on the 20 the network hasn't seen before.

Some other files have been changed to update python dependencies and error which followed.

(left is Original and right is Reconstructed) conv_mnist