├── .gitignore ├── LICENSE ├── README.md ├── figs ├── arch_detail.png ├── l3c_code_outline.jpg └── teaser.png ├── pip_requirements.txt └── src ├── auto_crop.py ├── bitcoding ├── __init__.py ├── bitcoding.py ├── coders.py ├── coders_helpers.py └── part_suffix_helper.py ├── blueprints ├── __init__.py └── multiscale_blueprint.py ├── configs ├── dl │ ├── in32.cf │ ├── in64.cf │ └── oi.cf └── ms │ ├── cr.cf │ ├── cr_rgb.cf │ └── cr_rgb_shared.cf ├── criterion ├── __init__.py └── logistic_mixture.py ├── dataloaders ├── __init__.py └── images_loader.py ├── helpers ├── __init__.py ├── aligned_printer.py ├── config_checker.py ├── global_config.py ├── logdir_helpers.py ├── pad.py ├── paths.py ├── rolling_buffer.py ├── saver.py └── testset.py ├── import_train_images.py ├── import_train_images_v1.py ├── l3c.py ├── modules ├── __init__.py ├── edsr.py ├── head.py ├── multiscale_network.py ├── net.py ├── prob_clf.py └── quantizer.py ├── prep_openimages.sh ├── pytorch_ext.py ├── test.py ├── test ├── __init__.py ├── cuda_timer.py ├── image_saver.py └── multiscale_tester.py ├── tmp_discard_list ├── torchac ├── setup.py ├── torchac.py └── torchac_backend │ ├── torchac.cpp │ └── torchac_kernel.cu ├── train.py ├── train ├── __init__.py ├── lr_schedule.py ├── multiscale_trainer.py ├── train_restorer.py └── trainer.py └── vis ├── __init__.py ├── figure_plotter.py ├── grid.py ├── histogram_plot.py ├── histogram_plotter.py ├── image_summaries.py ├── safe_summary_writer.py └── summarizable_module.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fab-jul/L3C-PyTorch/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fab-jul/L3C-PyTorch/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fab-jul/L3C-PyTorch/HEAD/README.md -------------------------------------------------------------------------------- /figs/arch_detail.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fab-jul/L3C-PyTorch/HEAD/figs/arch_detail.png -------------------------------------------------------------------------------- /figs/l3c_code_outline.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fab-jul/L3C-PyTorch/HEAD/figs/l3c_code_outline.jpg -------------------------------------------------------------------------------- /figs/teaser.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fab-jul/L3C-PyTorch/HEAD/figs/teaser.png -------------------------------------------------------------------------------- /pip_requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fab-jul/L3C-PyTorch/HEAD/pip_requirements.txt -------------------------------------------------------------------------------- /src/auto_crop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fab-jul/L3C-PyTorch/HEAD/src/auto_crop.py -------------------------------------------------------------------------------- /src/bitcoding/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/bitcoding/bitcoding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fab-jul/L3C-PyTorch/HEAD/src/bitcoding/bitcoding.py -------------------------------------------------------------------------------- /src/bitcoding/coders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fab-jul/L3C-PyTorch/HEAD/src/bitcoding/coders.py -------------------------------------------------------------------------------- /src/bitcoding/coders_helpers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fab-jul/L3C-PyTorch/HEAD/src/bitcoding/coders_helpers.py -------------------------------------------------------------------------------- /src/bitcoding/part_suffix_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fab-jul/L3C-PyTorch/HEAD/src/bitcoding/part_suffix_helper.py -------------------------------------------------------------------------------- /src/blueprints/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/blueprints/multiscale_blueprint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fab-jul/L3C-PyTorch/HEAD/src/blueprints/multiscale_blueprint.py -------------------------------------------------------------------------------- /src/configs/dl/in32.cf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fab-jul/L3C-PyTorch/HEAD/src/configs/dl/in32.cf -------------------------------------------------------------------------------- /src/configs/dl/in64.cf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fab-jul/L3C-PyTorch/HEAD/src/configs/dl/in64.cf -------------------------------------------------------------------------------- /src/configs/dl/oi.cf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fab-jul/L3C-PyTorch/HEAD/src/configs/dl/oi.cf -------------------------------------------------------------------------------- /src/configs/ms/cr.cf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fab-jul/L3C-PyTorch/HEAD/src/configs/ms/cr.cf -------------------------------------------------------------------------------- /src/configs/ms/cr_rgb.cf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fab-jul/L3C-PyTorch/HEAD/src/configs/ms/cr_rgb.cf -------------------------------------------------------------------------------- /src/configs/ms/cr_rgb_shared.cf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fab-jul/L3C-PyTorch/HEAD/src/configs/ms/cr_rgb_shared.cf -------------------------------------------------------------------------------- /src/criterion/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/criterion/logistic_mixture.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fab-jul/L3C-PyTorch/HEAD/src/criterion/logistic_mixture.py -------------------------------------------------------------------------------- /src/dataloaders/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/dataloaders/images_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fab-jul/L3C-PyTorch/HEAD/src/dataloaders/images_loader.py -------------------------------------------------------------------------------- /src/helpers/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/helpers/aligned_printer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fab-jul/L3C-PyTorch/HEAD/src/helpers/aligned_printer.py -------------------------------------------------------------------------------- /src/helpers/config_checker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fab-jul/L3C-PyTorch/HEAD/src/helpers/config_checker.py -------------------------------------------------------------------------------- /src/helpers/global_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fab-jul/L3C-PyTorch/HEAD/src/helpers/global_config.py -------------------------------------------------------------------------------- /src/helpers/logdir_helpers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fab-jul/L3C-PyTorch/HEAD/src/helpers/logdir_helpers.py -------------------------------------------------------------------------------- /src/helpers/pad.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fab-jul/L3C-PyTorch/HEAD/src/helpers/pad.py -------------------------------------------------------------------------------- /src/helpers/paths.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fab-jul/L3C-PyTorch/HEAD/src/helpers/paths.py -------------------------------------------------------------------------------- /src/helpers/rolling_buffer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fab-jul/L3C-PyTorch/HEAD/src/helpers/rolling_buffer.py -------------------------------------------------------------------------------- /src/helpers/saver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fab-jul/L3C-PyTorch/HEAD/src/helpers/saver.py -------------------------------------------------------------------------------- /src/helpers/testset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fab-jul/L3C-PyTorch/HEAD/src/helpers/testset.py -------------------------------------------------------------------------------- /src/import_train_images.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fab-jul/L3C-PyTorch/HEAD/src/import_train_images.py -------------------------------------------------------------------------------- /src/import_train_images_v1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fab-jul/L3C-PyTorch/HEAD/src/import_train_images_v1.py -------------------------------------------------------------------------------- /src/l3c.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fab-jul/L3C-PyTorch/HEAD/src/l3c.py -------------------------------------------------------------------------------- /src/modules/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/modules/edsr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fab-jul/L3C-PyTorch/HEAD/src/modules/edsr.py -------------------------------------------------------------------------------- /src/modules/head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fab-jul/L3C-PyTorch/HEAD/src/modules/head.py -------------------------------------------------------------------------------- /src/modules/multiscale_network.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fab-jul/L3C-PyTorch/HEAD/src/modules/multiscale_network.py -------------------------------------------------------------------------------- /src/modules/net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fab-jul/L3C-PyTorch/HEAD/src/modules/net.py -------------------------------------------------------------------------------- /src/modules/prob_clf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fab-jul/L3C-PyTorch/HEAD/src/modules/prob_clf.py -------------------------------------------------------------------------------- /src/modules/quantizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fab-jul/L3C-PyTorch/HEAD/src/modules/quantizer.py -------------------------------------------------------------------------------- /src/prep_openimages.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fab-jul/L3C-PyTorch/HEAD/src/prep_openimages.sh -------------------------------------------------------------------------------- /src/pytorch_ext.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fab-jul/L3C-PyTorch/HEAD/src/pytorch_ext.py -------------------------------------------------------------------------------- /src/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fab-jul/L3C-PyTorch/HEAD/src/test.py -------------------------------------------------------------------------------- /src/test/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/test/cuda_timer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fab-jul/L3C-PyTorch/HEAD/src/test/cuda_timer.py -------------------------------------------------------------------------------- /src/test/image_saver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fab-jul/L3C-PyTorch/HEAD/src/test/image_saver.py -------------------------------------------------------------------------------- /src/test/multiscale_tester.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fab-jul/L3C-PyTorch/HEAD/src/test/multiscale_tester.py -------------------------------------------------------------------------------- /src/tmp_discard_list: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fab-jul/L3C-PyTorch/HEAD/src/tmp_discard_list -------------------------------------------------------------------------------- /src/torchac/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fab-jul/L3C-PyTorch/HEAD/src/torchac/setup.py -------------------------------------------------------------------------------- /src/torchac/torchac.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fab-jul/L3C-PyTorch/HEAD/src/torchac/torchac.py -------------------------------------------------------------------------------- /src/torchac/torchac_backend/torchac.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fab-jul/L3C-PyTorch/HEAD/src/torchac/torchac_backend/torchac.cpp -------------------------------------------------------------------------------- /src/torchac/torchac_backend/torchac_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fab-jul/L3C-PyTorch/HEAD/src/torchac/torchac_backend/torchac_kernel.cu -------------------------------------------------------------------------------- /src/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fab-jul/L3C-PyTorch/HEAD/src/train.py -------------------------------------------------------------------------------- /src/train/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/train/lr_schedule.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fab-jul/L3C-PyTorch/HEAD/src/train/lr_schedule.py -------------------------------------------------------------------------------- /src/train/multiscale_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fab-jul/L3C-PyTorch/HEAD/src/train/multiscale_trainer.py -------------------------------------------------------------------------------- /src/train/train_restorer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fab-jul/L3C-PyTorch/HEAD/src/train/train_restorer.py -------------------------------------------------------------------------------- /src/train/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fab-jul/L3C-PyTorch/HEAD/src/train/trainer.py -------------------------------------------------------------------------------- /src/vis/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/vis/figure_plotter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fab-jul/L3C-PyTorch/HEAD/src/vis/figure_plotter.py -------------------------------------------------------------------------------- /src/vis/grid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fab-jul/L3C-PyTorch/HEAD/src/vis/grid.py -------------------------------------------------------------------------------- /src/vis/histogram_plot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fab-jul/L3C-PyTorch/HEAD/src/vis/histogram_plot.py -------------------------------------------------------------------------------- /src/vis/histogram_plotter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fab-jul/L3C-PyTorch/HEAD/src/vis/histogram_plotter.py -------------------------------------------------------------------------------- /src/vis/image_summaries.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fab-jul/L3C-PyTorch/HEAD/src/vis/image_summaries.py -------------------------------------------------------------------------------- /src/vis/safe_summary_writer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fab-jul/L3C-PyTorch/HEAD/src/vis/safe_summary_writer.py -------------------------------------------------------------------------------- /src/vis/summarizable_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fab-jul/L3C-PyTorch/HEAD/src/vis/summarizable_module.py --------------------------------------------------------------------------------