├── .gitignore ├── LICENSE ├── README.md ├── code ├── .gitignore ├── ae_configs │ ├── base │ └── cvpr │ │ ├── base │ │ ├── hi │ │ ├── low │ │ └── med ├── arithmetic_coding.py ├── autoencoder.py ├── bit_counter.py ├── bits.py ├── bpp_helpers.py ├── codec_distance.py ├── compare_imgs.py ├── constants.py ├── images_iterator.py ├── inputpipeline.py ├── logdir_helpers.py ├── logger.py ├── ms_ssim.py ├── ms_ssim_np.py ├── other_codecs.py ├── pc_configs │ ├── base │ └── cvpr │ │ ├── res_shallow │ │ └── res_shallow_64 ├── plotter.py ├── probclass.py ├── purge_checkpoints.py ├── quantizer.py ├── restore_manager.py ├── save_mapper.py ├── saver.py ├── sheets_logger.py ├── train.py ├── training_helpers.py ├── val.py ├── val_files.py └── val_images.py ├── figs ├── plot_Kodak.png └── teaser.jpg └── requirements.txt /.gitignore: -------------------------------------------------------------------------------- 1 | __pycache__/ 2 | 3 | .DS_Store 4 | .idea/ 5 | cdvl/ 6 | *.tfrecords 7 | 8 | -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fab-jul/imgcomp-cvpr/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fab-jul/imgcomp-cvpr/HEAD/README.md -------------------------------------------------------------------------------- /code/.gitignore: -------------------------------------------------------------------------------- 1 | __pycache__/ 2 | 3 | .DS_Store 4 | .idea/ 5 | cdvl/ 6 | *.tfrecords 7 | 8 | -------------------------------------------------------------------------------- /code/ae_configs/base: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fab-jul/imgcomp-cvpr/HEAD/code/ae_configs/base -------------------------------------------------------------------------------- /code/ae_configs/cvpr/base: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fab-jul/imgcomp-cvpr/HEAD/code/ae_configs/cvpr/base -------------------------------------------------------------------------------- /code/ae_configs/cvpr/hi: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fab-jul/imgcomp-cvpr/HEAD/code/ae_configs/cvpr/hi -------------------------------------------------------------------------------- /code/ae_configs/cvpr/low: -------------------------------------------------------------------------------- 1 | use base 2 | 3 | num_chan_bn = 32 4 | H_target = 2*0.2 # == 64/C * bpp 5 | 6 | 7 | -------------------------------------------------------------------------------- /code/ae_configs/cvpr/med: -------------------------------------------------------------------------------- 1 | use base 2 | 3 | num_chan_bn = 32 4 | H_target = 2*0.6 # == 64/C * bpp 5 | 6 | 7 | -------------------------------------------------------------------------------- /code/arithmetic_coding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fab-jul/imgcomp-cvpr/HEAD/code/arithmetic_coding.py -------------------------------------------------------------------------------- /code/autoencoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fab-jul/imgcomp-cvpr/HEAD/code/autoencoder.py -------------------------------------------------------------------------------- /code/bit_counter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fab-jul/imgcomp-cvpr/HEAD/code/bit_counter.py -------------------------------------------------------------------------------- /code/bits.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fab-jul/imgcomp-cvpr/HEAD/code/bits.py -------------------------------------------------------------------------------- /code/bpp_helpers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fab-jul/imgcomp-cvpr/HEAD/code/bpp_helpers.py -------------------------------------------------------------------------------- /code/codec_distance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fab-jul/imgcomp-cvpr/HEAD/code/codec_distance.py -------------------------------------------------------------------------------- /code/compare_imgs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fab-jul/imgcomp-cvpr/HEAD/code/compare_imgs.py -------------------------------------------------------------------------------- /code/constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fab-jul/imgcomp-cvpr/HEAD/code/constants.py -------------------------------------------------------------------------------- /code/images_iterator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fab-jul/imgcomp-cvpr/HEAD/code/images_iterator.py -------------------------------------------------------------------------------- /code/inputpipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fab-jul/imgcomp-cvpr/HEAD/code/inputpipeline.py -------------------------------------------------------------------------------- /code/logdir_helpers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fab-jul/imgcomp-cvpr/HEAD/code/logdir_helpers.py -------------------------------------------------------------------------------- /code/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fab-jul/imgcomp-cvpr/HEAD/code/logger.py -------------------------------------------------------------------------------- /code/ms_ssim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fab-jul/imgcomp-cvpr/HEAD/code/ms_ssim.py -------------------------------------------------------------------------------- /code/ms_ssim_np.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fab-jul/imgcomp-cvpr/HEAD/code/ms_ssim_np.py -------------------------------------------------------------------------------- /code/other_codecs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fab-jul/imgcomp-cvpr/HEAD/code/other_codecs.py -------------------------------------------------------------------------------- /code/pc_configs/base: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fab-jul/imgcomp-cvpr/HEAD/code/pc_configs/base -------------------------------------------------------------------------------- /code/pc_configs/cvpr/res_shallow: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fab-jul/imgcomp-cvpr/HEAD/code/pc_configs/cvpr/res_shallow -------------------------------------------------------------------------------- /code/pc_configs/cvpr/res_shallow_64: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fab-jul/imgcomp-cvpr/HEAD/code/pc_configs/cvpr/res_shallow_64 -------------------------------------------------------------------------------- /code/plotter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fab-jul/imgcomp-cvpr/HEAD/code/plotter.py -------------------------------------------------------------------------------- /code/probclass.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fab-jul/imgcomp-cvpr/HEAD/code/probclass.py -------------------------------------------------------------------------------- /code/purge_checkpoints.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fab-jul/imgcomp-cvpr/HEAD/code/purge_checkpoints.py -------------------------------------------------------------------------------- /code/quantizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fab-jul/imgcomp-cvpr/HEAD/code/quantizer.py -------------------------------------------------------------------------------- /code/restore_manager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fab-jul/imgcomp-cvpr/HEAD/code/restore_manager.py -------------------------------------------------------------------------------- /code/save_mapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fab-jul/imgcomp-cvpr/HEAD/code/save_mapper.py -------------------------------------------------------------------------------- /code/saver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fab-jul/imgcomp-cvpr/HEAD/code/saver.py -------------------------------------------------------------------------------- /code/sheets_logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fab-jul/imgcomp-cvpr/HEAD/code/sheets_logger.py -------------------------------------------------------------------------------- /code/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fab-jul/imgcomp-cvpr/HEAD/code/train.py -------------------------------------------------------------------------------- /code/training_helpers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fab-jul/imgcomp-cvpr/HEAD/code/training_helpers.py -------------------------------------------------------------------------------- /code/val.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fab-jul/imgcomp-cvpr/HEAD/code/val.py -------------------------------------------------------------------------------- /code/val_files.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fab-jul/imgcomp-cvpr/HEAD/code/val_files.py -------------------------------------------------------------------------------- /code/val_images.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fab-jul/imgcomp-cvpr/HEAD/code/val_images.py -------------------------------------------------------------------------------- /figs/plot_Kodak.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fab-jul/imgcomp-cvpr/HEAD/figs/plot_Kodak.png -------------------------------------------------------------------------------- /figs/teaser.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fab-jul/imgcomp-cvpr/HEAD/figs/teaser.jpg -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fab-jul/imgcomp-cvpr/HEAD/requirements.txt --------------------------------------------------------------------------------