├── .gitignore ├── LICENSE ├── README.md ├── assets ├── clic_rd.png ├── cnn.png ├── cnn_arch.pdf ├── cnn_arch.png ├── detail_01.png ├── detail_07.png ├── kodak_rd.png └── stf_arch.png ├── compressai ├── __init__.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 │ └── win_attention.py ├── models │ ├── __init__.py │ ├── base.py │ ├── cnn.py │ ├── stf.py │ └── utils.py ├── ops │ ├── __init__.py │ ├── bound_ops.py │ ├── ops.py │ └── parametrizers.py ├── utils │ ├── __init__.py │ └── eval_model │ │ ├── __init__.py │ │ └── __main__.py ├── version.py └── zoo │ ├── __init__.py │ └── pretrained.py ├── downloader_openimages.py ├── requirements.txt ├── results ├── cnn_ms-ssim_CLIC - .json ├── cnn_ms-ssim_Kodak - .json ├── cnn_mse_CLIC.json ├── cnn_mse_Kodak.json ├── stf_ms-ssim_CLIC.json ├── stf_ms-ssim_Kodak.json ├── stf_mse_CLIC .json └── stf_mse_Kodak.json ├── setup.py ├── third_party └── ryg_rans │ ├── LICENSE │ ├── README │ ├── rans64.h │ ├── rans_byte.h │ └── rans_word_sse41.h └── train.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Googolxx/STF/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Googolxx/STF/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Googolxx/STF/HEAD/README.md -------------------------------------------------------------------------------- /assets/clic_rd.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Googolxx/STF/HEAD/assets/clic_rd.png -------------------------------------------------------------------------------- /assets/cnn.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Googolxx/STF/HEAD/assets/cnn.png -------------------------------------------------------------------------------- /assets/cnn_arch.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Googolxx/STF/HEAD/assets/cnn_arch.pdf -------------------------------------------------------------------------------- /assets/cnn_arch.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Googolxx/STF/HEAD/assets/cnn_arch.png -------------------------------------------------------------------------------- /assets/detail_01.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Googolxx/STF/HEAD/assets/detail_01.png -------------------------------------------------------------------------------- /assets/detail_07.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Googolxx/STF/HEAD/assets/detail_07.png -------------------------------------------------------------------------------- /assets/kodak_rd.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Googolxx/STF/HEAD/assets/kodak_rd.png -------------------------------------------------------------------------------- /assets/stf_arch.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Googolxx/STF/HEAD/assets/stf_arch.png -------------------------------------------------------------------------------- /compressai/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Googolxx/STF/HEAD/compressai/__init__.py -------------------------------------------------------------------------------- /compressai/cpp_exts/ops/ops.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Googolxx/STF/HEAD/compressai/cpp_exts/ops/ops.cpp -------------------------------------------------------------------------------- /compressai/cpp_exts/rans/rans_interface.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Googolxx/STF/HEAD/compressai/cpp_exts/rans/rans_interface.cpp -------------------------------------------------------------------------------- /compressai/cpp_exts/rans/rans_interface.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Googolxx/STF/HEAD/compressai/cpp_exts/rans/rans_interface.hpp -------------------------------------------------------------------------------- /compressai/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Googolxx/STF/HEAD/compressai/datasets/__init__.py -------------------------------------------------------------------------------- /compressai/datasets/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Googolxx/STF/HEAD/compressai/datasets/utils.py -------------------------------------------------------------------------------- /compressai/entropy_models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Googolxx/STF/HEAD/compressai/entropy_models/__init__.py -------------------------------------------------------------------------------- /compressai/entropy_models/entropy_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Googolxx/STF/HEAD/compressai/entropy_models/entropy_models.py -------------------------------------------------------------------------------- /compressai/layers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Googolxx/STF/HEAD/compressai/layers/__init__.py -------------------------------------------------------------------------------- /compressai/layers/gdn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Googolxx/STF/HEAD/compressai/layers/gdn.py -------------------------------------------------------------------------------- /compressai/layers/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Googolxx/STF/HEAD/compressai/layers/layers.py -------------------------------------------------------------------------------- /compressai/layers/win_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Googolxx/STF/HEAD/compressai/layers/win_attention.py -------------------------------------------------------------------------------- /compressai/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Googolxx/STF/HEAD/compressai/models/__init__.py -------------------------------------------------------------------------------- /compressai/models/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Googolxx/STF/HEAD/compressai/models/base.py -------------------------------------------------------------------------------- /compressai/models/cnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Googolxx/STF/HEAD/compressai/models/cnn.py -------------------------------------------------------------------------------- /compressai/models/stf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Googolxx/STF/HEAD/compressai/models/stf.py -------------------------------------------------------------------------------- /compressai/models/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Googolxx/STF/HEAD/compressai/models/utils.py -------------------------------------------------------------------------------- /compressai/ops/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Googolxx/STF/HEAD/compressai/ops/__init__.py -------------------------------------------------------------------------------- /compressai/ops/bound_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Googolxx/STF/HEAD/compressai/ops/bound_ops.py -------------------------------------------------------------------------------- /compressai/ops/ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Googolxx/STF/HEAD/compressai/ops/ops.py -------------------------------------------------------------------------------- /compressai/ops/parametrizers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Googolxx/STF/HEAD/compressai/ops/parametrizers.py -------------------------------------------------------------------------------- /compressai/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Googolxx/STF/HEAD/compressai/utils/__init__.py -------------------------------------------------------------------------------- /compressai/utils/eval_model/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Googolxx/STF/HEAD/compressai/utils/eval_model/__init__.py -------------------------------------------------------------------------------- /compressai/utils/eval_model/__main__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Googolxx/STF/HEAD/compressai/utils/eval_model/__main__.py -------------------------------------------------------------------------------- /compressai/version.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Googolxx/STF/HEAD/compressai/version.py -------------------------------------------------------------------------------- /compressai/zoo/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Googolxx/STF/HEAD/compressai/zoo/__init__.py -------------------------------------------------------------------------------- /compressai/zoo/pretrained.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Googolxx/STF/HEAD/compressai/zoo/pretrained.py -------------------------------------------------------------------------------- /downloader_openimages.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Googolxx/STF/HEAD/downloader_openimages.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Googolxx/STF/HEAD/requirements.txt -------------------------------------------------------------------------------- /results/cnn_ms-ssim_CLIC - .json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Googolxx/STF/HEAD/results/cnn_ms-ssim_CLIC - .json -------------------------------------------------------------------------------- /results/cnn_ms-ssim_Kodak - .json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Googolxx/STF/HEAD/results/cnn_ms-ssim_Kodak - .json -------------------------------------------------------------------------------- /results/cnn_mse_CLIC.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Googolxx/STF/HEAD/results/cnn_mse_CLIC.json -------------------------------------------------------------------------------- /results/cnn_mse_Kodak.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Googolxx/STF/HEAD/results/cnn_mse_Kodak.json -------------------------------------------------------------------------------- /results/stf_ms-ssim_CLIC.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Googolxx/STF/HEAD/results/stf_ms-ssim_CLIC.json -------------------------------------------------------------------------------- /results/stf_ms-ssim_Kodak.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Googolxx/STF/HEAD/results/stf_ms-ssim_Kodak.json -------------------------------------------------------------------------------- /results/stf_mse_CLIC .json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Googolxx/STF/HEAD/results/stf_mse_CLIC .json -------------------------------------------------------------------------------- /results/stf_mse_Kodak.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Googolxx/STF/HEAD/results/stf_mse_Kodak.json -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Googolxx/STF/HEAD/setup.py -------------------------------------------------------------------------------- /third_party/ryg_rans/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Googolxx/STF/HEAD/third_party/ryg_rans/LICENSE -------------------------------------------------------------------------------- /third_party/ryg_rans/README: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Googolxx/STF/HEAD/third_party/ryg_rans/README -------------------------------------------------------------------------------- /third_party/ryg_rans/rans64.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Googolxx/STF/HEAD/third_party/ryg_rans/rans64.h -------------------------------------------------------------------------------- /third_party/ryg_rans/rans_byte.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Googolxx/STF/HEAD/third_party/ryg_rans/rans_byte.h -------------------------------------------------------------------------------- /third_party/ryg_rans/rans_word_sse41.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Googolxx/STF/HEAD/third_party/ryg_rans/rans_word_sse41.h -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Googolxx/STF/HEAD/train.py --------------------------------------------------------------------------------