├── .idea ├── .gitignore ├── GainedVAE.iml ├── inspectionProfiles │ ├── Project_Default.xml │ └── profiles_settings.xml ├── misc.xml ├── modules.xml └── vcs.xml ├── README.md ├── compressai ├── _CXX.cp37-win_amd64.pyd ├── _CXX.py ├── __init__.py ├── ans.cp37-win_amd64.pyd ├── ans.py ├── cpp_exts │ ├── ops │ │ └── ops.cpp │ └── rans │ │ ├── rans_interface.cpp │ │ └── rans_interface.hpp ├── datasets │ ├── __init__.py │ └── utils.py ├── entropy_models │ ├── __init__.py │ └── entropy_models.py ├── layers │ ├── __init__.py │ ├── gdn.py │ └── layers.py ├── models │ ├── __init__.py │ ├── gain.py │ ├── gain_utils.py │ ├── priors.py │ ├── utils.py │ └── waseda.py ├── ops │ ├── __init__.py │ ├── bound_ops.py │ ├── ops.py │ └── parametrizers.py ├── transforms │ ├── __init__.py │ ├── functional.py │ └── transforms.py ├── utils │ ├── __init__.py │ ├── bench │ │ ├── __init__.py │ │ ├── __main__.py │ │ └── codecs.py │ ├── eval_model │ │ ├── __init__.py │ │ └── __main__.py │ ├── find_close │ │ ├── __init__.py │ │ └── __main__.py │ ├── plot │ │ ├── __init__.py │ │ └── __main__.py │ └── update_model │ │ ├── __init__.py │ │ └── __main__.py ├── version.py └── zoo │ ├── __init__.py │ ├── image.py │ └── pretrained.py ├── compressai_examples ├── assets │ └── stmalo_fracape.png ├── codec.py └── train.py ├── requirements.txt ├── results ├── MSHyperprior Results.txt ├── ScaleHyperpriors Results.txt ├── mshyper.png └── scalehyper.png ├── setup.py ├── tests ├── assets │ └── fakedata │ │ └── imagefolder │ │ ├── test │ │ └── logo.png │ │ └── train │ │ └── logo.png ├── test_codec.py ├── test_coder.py ├── test_datasets.py ├── test_entropy_models.py ├── test_init.py ├── test_layers.py ├── test_models.py ├── test_ops.py ├── test_scripting.py ├── test_train.py ├── test_transforms.py ├── test_update_model.py ├── test_waseda.py └── test_zoo.py ├── third_party └── ryg_rans │ ├── LICENSE │ ├── README │ ├── rans64.h │ ├── rans_byte.h │ └── rans_word_sse41.h └── train ├── eval_gain.py ├── roi_image_dataset.py ├── trainGain.py └── train_scgain.py /.idea/.gitignore: -------------------------------------------------------------------------------- 1 | # Default ignored files 2 | /shelf/ 3 | /workspace.xml 4 | -------------------------------------------------------------------------------- /.idea/GainedVAE.iml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmSir/GainedVAE/HEAD/.idea/GainedVAE.iml -------------------------------------------------------------------------------- /.idea/inspectionProfiles/Project_Default.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmSir/GainedVAE/HEAD/.idea/inspectionProfiles/Project_Default.xml -------------------------------------------------------------------------------- /.idea/inspectionProfiles/profiles_settings.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmSir/GainedVAE/HEAD/.idea/inspectionProfiles/profiles_settings.xml -------------------------------------------------------------------------------- /.idea/misc.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmSir/GainedVAE/HEAD/.idea/misc.xml -------------------------------------------------------------------------------- /.idea/modules.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmSir/GainedVAE/HEAD/.idea/modules.xml -------------------------------------------------------------------------------- /.idea/vcs.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmSir/GainedVAE/HEAD/.idea/vcs.xml -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmSir/GainedVAE/HEAD/README.md -------------------------------------------------------------------------------- /compressai/_CXX.cp37-win_amd64.pyd: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmSir/GainedVAE/HEAD/compressai/_CXX.cp37-win_amd64.pyd -------------------------------------------------------------------------------- /compressai/_CXX.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmSir/GainedVAE/HEAD/compressai/_CXX.py -------------------------------------------------------------------------------- /compressai/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmSir/GainedVAE/HEAD/compressai/__init__.py -------------------------------------------------------------------------------- /compressai/ans.cp37-win_amd64.pyd: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmSir/GainedVAE/HEAD/compressai/ans.cp37-win_amd64.pyd -------------------------------------------------------------------------------- /compressai/ans.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmSir/GainedVAE/HEAD/compressai/ans.py -------------------------------------------------------------------------------- /compressai/cpp_exts/ops/ops.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmSir/GainedVAE/HEAD/compressai/cpp_exts/ops/ops.cpp -------------------------------------------------------------------------------- /compressai/cpp_exts/rans/rans_interface.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmSir/GainedVAE/HEAD/compressai/cpp_exts/rans/rans_interface.cpp -------------------------------------------------------------------------------- /compressai/cpp_exts/rans/rans_interface.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmSir/GainedVAE/HEAD/compressai/cpp_exts/rans/rans_interface.hpp -------------------------------------------------------------------------------- /compressai/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmSir/GainedVAE/HEAD/compressai/datasets/__init__.py -------------------------------------------------------------------------------- /compressai/datasets/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmSir/GainedVAE/HEAD/compressai/datasets/utils.py -------------------------------------------------------------------------------- /compressai/entropy_models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmSir/GainedVAE/HEAD/compressai/entropy_models/__init__.py -------------------------------------------------------------------------------- /compressai/entropy_models/entropy_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmSir/GainedVAE/HEAD/compressai/entropy_models/entropy_models.py -------------------------------------------------------------------------------- /compressai/layers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmSir/GainedVAE/HEAD/compressai/layers/__init__.py -------------------------------------------------------------------------------- /compressai/layers/gdn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmSir/GainedVAE/HEAD/compressai/layers/gdn.py -------------------------------------------------------------------------------- /compressai/layers/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmSir/GainedVAE/HEAD/compressai/layers/layers.py -------------------------------------------------------------------------------- /compressai/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmSir/GainedVAE/HEAD/compressai/models/__init__.py -------------------------------------------------------------------------------- /compressai/models/gain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmSir/GainedVAE/HEAD/compressai/models/gain.py -------------------------------------------------------------------------------- /compressai/models/gain_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmSir/GainedVAE/HEAD/compressai/models/gain_utils.py -------------------------------------------------------------------------------- /compressai/models/priors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmSir/GainedVAE/HEAD/compressai/models/priors.py -------------------------------------------------------------------------------- /compressai/models/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmSir/GainedVAE/HEAD/compressai/models/utils.py -------------------------------------------------------------------------------- /compressai/models/waseda.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmSir/GainedVAE/HEAD/compressai/models/waseda.py -------------------------------------------------------------------------------- /compressai/ops/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmSir/GainedVAE/HEAD/compressai/ops/__init__.py -------------------------------------------------------------------------------- /compressai/ops/bound_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmSir/GainedVAE/HEAD/compressai/ops/bound_ops.py -------------------------------------------------------------------------------- /compressai/ops/ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmSir/GainedVAE/HEAD/compressai/ops/ops.py -------------------------------------------------------------------------------- /compressai/ops/parametrizers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmSir/GainedVAE/HEAD/compressai/ops/parametrizers.py -------------------------------------------------------------------------------- /compressai/transforms/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmSir/GainedVAE/HEAD/compressai/transforms/__init__.py -------------------------------------------------------------------------------- /compressai/transforms/functional.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmSir/GainedVAE/HEAD/compressai/transforms/functional.py -------------------------------------------------------------------------------- /compressai/transforms/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmSir/GainedVAE/HEAD/compressai/transforms/transforms.py -------------------------------------------------------------------------------- /compressai/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmSir/GainedVAE/HEAD/compressai/utils/__init__.py -------------------------------------------------------------------------------- /compressai/utils/bench/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmSir/GainedVAE/HEAD/compressai/utils/bench/__init__.py -------------------------------------------------------------------------------- /compressai/utils/bench/__main__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmSir/GainedVAE/HEAD/compressai/utils/bench/__main__.py -------------------------------------------------------------------------------- /compressai/utils/bench/codecs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmSir/GainedVAE/HEAD/compressai/utils/bench/codecs.py -------------------------------------------------------------------------------- /compressai/utils/eval_model/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmSir/GainedVAE/HEAD/compressai/utils/eval_model/__init__.py -------------------------------------------------------------------------------- /compressai/utils/eval_model/__main__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmSir/GainedVAE/HEAD/compressai/utils/eval_model/__main__.py -------------------------------------------------------------------------------- /compressai/utils/find_close/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /compressai/utils/find_close/__main__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmSir/GainedVAE/HEAD/compressai/utils/find_close/__main__.py -------------------------------------------------------------------------------- /compressai/utils/plot/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmSir/GainedVAE/HEAD/compressai/utils/plot/__init__.py -------------------------------------------------------------------------------- /compressai/utils/plot/__main__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmSir/GainedVAE/HEAD/compressai/utils/plot/__main__.py -------------------------------------------------------------------------------- /compressai/utils/update_model/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmSir/GainedVAE/HEAD/compressai/utils/update_model/__init__.py -------------------------------------------------------------------------------- /compressai/utils/update_model/__main__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmSir/GainedVAE/HEAD/compressai/utils/update_model/__main__.py -------------------------------------------------------------------------------- /compressai/version.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmSir/GainedVAE/HEAD/compressai/version.py -------------------------------------------------------------------------------- /compressai/zoo/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmSir/GainedVAE/HEAD/compressai/zoo/__init__.py -------------------------------------------------------------------------------- /compressai/zoo/image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmSir/GainedVAE/HEAD/compressai/zoo/image.py -------------------------------------------------------------------------------- /compressai/zoo/pretrained.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmSir/GainedVAE/HEAD/compressai/zoo/pretrained.py -------------------------------------------------------------------------------- /compressai_examples/assets/stmalo_fracape.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmSir/GainedVAE/HEAD/compressai_examples/assets/stmalo_fracape.png -------------------------------------------------------------------------------- /compressai_examples/codec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmSir/GainedVAE/HEAD/compressai_examples/codec.py -------------------------------------------------------------------------------- /compressai_examples/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmSir/GainedVAE/HEAD/compressai_examples/train.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmSir/GainedVAE/HEAD/requirements.txt -------------------------------------------------------------------------------- /results/MSHyperprior Results.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmSir/GainedVAE/HEAD/results/MSHyperprior Results.txt -------------------------------------------------------------------------------- /results/ScaleHyperpriors Results.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmSir/GainedVAE/HEAD/results/ScaleHyperpriors Results.txt -------------------------------------------------------------------------------- /results/mshyper.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmSir/GainedVAE/HEAD/results/mshyper.png -------------------------------------------------------------------------------- /results/scalehyper.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmSir/GainedVAE/HEAD/results/scalehyper.png -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmSir/GainedVAE/HEAD/setup.py -------------------------------------------------------------------------------- /tests/assets/fakedata/imagefolder/test/logo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmSir/GainedVAE/HEAD/tests/assets/fakedata/imagefolder/test/logo.png -------------------------------------------------------------------------------- /tests/assets/fakedata/imagefolder/train/logo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmSir/GainedVAE/HEAD/tests/assets/fakedata/imagefolder/train/logo.png -------------------------------------------------------------------------------- /tests/test_codec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmSir/GainedVAE/HEAD/tests/test_codec.py -------------------------------------------------------------------------------- /tests/test_coder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmSir/GainedVAE/HEAD/tests/test_coder.py -------------------------------------------------------------------------------- /tests/test_datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmSir/GainedVAE/HEAD/tests/test_datasets.py -------------------------------------------------------------------------------- /tests/test_entropy_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmSir/GainedVAE/HEAD/tests/test_entropy_models.py -------------------------------------------------------------------------------- /tests/test_init.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmSir/GainedVAE/HEAD/tests/test_init.py -------------------------------------------------------------------------------- /tests/test_layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmSir/GainedVAE/HEAD/tests/test_layers.py -------------------------------------------------------------------------------- /tests/test_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmSir/GainedVAE/HEAD/tests/test_models.py -------------------------------------------------------------------------------- /tests/test_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmSir/GainedVAE/HEAD/tests/test_ops.py -------------------------------------------------------------------------------- /tests/test_scripting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmSir/GainedVAE/HEAD/tests/test_scripting.py -------------------------------------------------------------------------------- /tests/test_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmSir/GainedVAE/HEAD/tests/test_train.py -------------------------------------------------------------------------------- /tests/test_transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmSir/GainedVAE/HEAD/tests/test_transforms.py -------------------------------------------------------------------------------- /tests/test_update_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmSir/GainedVAE/HEAD/tests/test_update_model.py -------------------------------------------------------------------------------- /tests/test_waseda.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmSir/GainedVAE/HEAD/tests/test_waseda.py -------------------------------------------------------------------------------- /tests/test_zoo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmSir/GainedVAE/HEAD/tests/test_zoo.py -------------------------------------------------------------------------------- /third_party/ryg_rans/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmSir/GainedVAE/HEAD/third_party/ryg_rans/LICENSE -------------------------------------------------------------------------------- /third_party/ryg_rans/README: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmSir/GainedVAE/HEAD/third_party/ryg_rans/README -------------------------------------------------------------------------------- /third_party/ryg_rans/rans64.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmSir/GainedVAE/HEAD/third_party/ryg_rans/rans64.h -------------------------------------------------------------------------------- /third_party/ryg_rans/rans_byte.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmSir/GainedVAE/HEAD/third_party/ryg_rans/rans_byte.h -------------------------------------------------------------------------------- /third_party/ryg_rans/rans_word_sse41.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmSir/GainedVAE/HEAD/third_party/ryg_rans/rans_word_sse41.h -------------------------------------------------------------------------------- /train/eval_gain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmSir/GainedVAE/HEAD/train/eval_gain.py -------------------------------------------------------------------------------- /train/roi_image_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmSir/GainedVAE/HEAD/train/roi_image_dataset.py -------------------------------------------------------------------------------- /train/trainGain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmSir/GainedVAE/HEAD/train/trainGain.py -------------------------------------------------------------------------------- /train/train_scgain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmSir/GainedVAE/HEAD/train/train_scgain.py --------------------------------------------------------------------------------