Skip to content

Generic image compressor for machine learning. Pytorch code for our paper "Lossy compression for lossless prediction".

License

Notifications You must be signed in to change notification settings

YannDubs/lossyless

Repository files navigation

Lossy Compression for Lossless Prediction License: MIT Python 3.6+

WIP

Install

  1. Clone repository
  2. Install PyTorch >= 1.7
  3. pip install -r requirements.txt

Nota Bene:

  • if you prefer I also provide a Dockerfile to install the necessary packages.
  • hydra and pytorch lightning logging don't work great together (specifically pytorch lighning logs go to stderr and don't propagate). To have a better logging experience you should comment (or delete) out the folowing lines in pytorch_lightning/__init__.py :
if not _root_logger.hasHandlers():
     _logger.addHandler(logging.StreamHandler())
     _logger.propagate = False