├── LICENSE ├── Makefile ├── README.md ├── compressai ├── __init__.py ├── cpp_exts │ ├── ops │ │ └── ops.cpp │ └── rans │ │ ├── rans_interface.cpp │ │ └── rans_interface.hpp ├── datasets │ ├── __init__.py │ ├── image.py │ ├── rawvideo.py │ └── video.py ├── entropy_models │ ├── __init__.py │ └── entropy_models.py ├── layers │ ├── __init__.py │ ├── gdn.py │ ├── layers.py │ └── natten │ │ ├── .DS_Store │ │ ├── __init__.py │ │ ├── extras.py │ │ ├── flops.py │ │ ├── gradcheck.py │ │ ├── gradcheck1d.py │ │ ├── nattencuda.py │ │ ├── nattencuda1d.py │ │ ├── nattentorch.py │ │ └── src │ │ ├── natten1dav_cuda.cpp │ │ ├── natten1dav_cuda_kernel.cu │ │ ├── natten1dqkrpb_cuda.cpp │ │ ├── natten1dqkrpb_cuda_kernel.cu │ │ ├── natten_commons.cuh │ │ ├── nattenav_cuda.cpp │ │ ├── nattenav_cuda_kernel.cu │ │ ├── nattenqkrpb_cuda.cpp │ │ ├── nattenqkrpb_cuda_kernel.cu │ │ └── setup.py ├── models │ ├── __init__.py │ ├── google.py │ ├── priors.py │ ├── tinylic.py │ ├── utils.py │ ├── video │ │ ├── __init__.py │ │ └── google.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 │ └── video │ │ ├── __init__.py │ │ ├── bench │ │ ├── __init__.py │ │ ├── __main__.py │ │ └── codecs.py │ │ ├── eval_model │ │ ├── __init__.py │ │ └── __main__.py │ │ └── plot │ │ ├── __init__.py │ │ └── __main__.py └── zoo │ ├── __init__.py │ ├── image.py │ ├── pretrained.py │ └── video.py ├── examples └── train.py ├── pyproject.toml ├── requirements.txt ├── results ├── ms-ssim │ ├── clic.json │ ├── kodak.json │ └── tecnick.json └── mse │ ├── clic.json │ ├── kodak.json │ └── tecnick.json ├── setup.py └── third_party └── ryg_rans ├── LICENSE ├── README ├── rans64.h ├── rans_byte.h └── rans_word_sse41.h /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lumingzzz/TinyLIC/HEAD/LICENSE -------------------------------------------------------------------------------- /Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lumingzzz/TinyLIC/HEAD/Makefile -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lumingzzz/TinyLIC/HEAD/README.md -------------------------------------------------------------------------------- /compressai/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lumingzzz/TinyLIC/HEAD/compressai/__init__.py -------------------------------------------------------------------------------- /compressai/cpp_exts/ops/ops.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lumingzzz/TinyLIC/HEAD/compressai/cpp_exts/ops/ops.cpp -------------------------------------------------------------------------------- /compressai/cpp_exts/rans/rans_interface.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lumingzzz/TinyLIC/HEAD/compressai/cpp_exts/rans/rans_interface.cpp -------------------------------------------------------------------------------- /compressai/cpp_exts/rans/rans_interface.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lumingzzz/TinyLIC/HEAD/compressai/cpp_exts/rans/rans_interface.hpp -------------------------------------------------------------------------------- /compressai/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lumingzzz/TinyLIC/HEAD/compressai/datasets/__init__.py -------------------------------------------------------------------------------- /compressai/datasets/image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lumingzzz/TinyLIC/HEAD/compressai/datasets/image.py -------------------------------------------------------------------------------- /compressai/datasets/rawvideo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lumingzzz/TinyLIC/HEAD/compressai/datasets/rawvideo.py -------------------------------------------------------------------------------- /compressai/datasets/video.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lumingzzz/TinyLIC/HEAD/compressai/datasets/video.py -------------------------------------------------------------------------------- /compressai/entropy_models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lumingzzz/TinyLIC/HEAD/compressai/entropy_models/__init__.py -------------------------------------------------------------------------------- /compressai/entropy_models/entropy_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lumingzzz/TinyLIC/HEAD/compressai/entropy_models/entropy_models.py -------------------------------------------------------------------------------- /compressai/layers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lumingzzz/TinyLIC/HEAD/compressai/layers/__init__.py -------------------------------------------------------------------------------- /compressai/layers/gdn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lumingzzz/TinyLIC/HEAD/compressai/layers/gdn.py -------------------------------------------------------------------------------- /compressai/layers/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lumingzzz/TinyLIC/HEAD/compressai/layers/layers.py -------------------------------------------------------------------------------- /compressai/layers/natten/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lumingzzz/TinyLIC/HEAD/compressai/layers/natten/.DS_Store -------------------------------------------------------------------------------- /compressai/layers/natten/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lumingzzz/TinyLIC/HEAD/compressai/layers/natten/__init__.py -------------------------------------------------------------------------------- /compressai/layers/natten/extras.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lumingzzz/TinyLIC/HEAD/compressai/layers/natten/extras.py -------------------------------------------------------------------------------- /compressai/layers/natten/flops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lumingzzz/TinyLIC/HEAD/compressai/layers/natten/flops.py -------------------------------------------------------------------------------- /compressai/layers/natten/gradcheck.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lumingzzz/TinyLIC/HEAD/compressai/layers/natten/gradcheck.py -------------------------------------------------------------------------------- /compressai/layers/natten/gradcheck1d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lumingzzz/TinyLIC/HEAD/compressai/layers/natten/gradcheck1d.py -------------------------------------------------------------------------------- /compressai/layers/natten/nattencuda.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lumingzzz/TinyLIC/HEAD/compressai/layers/natten/nattencuda.py -------------------------------------------------------------------------------- /compressai/layers/natten/nattencuda1d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lumingzzz/TinyLIC/HEAD/compressai/layers/natten/nattencuda1d.py -------------------------------------------------------------------------------- /compressai/layers/natten/nattentorch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lumingzzz/TinyLIC/HEAD/compressai/layers/natten/nattentorch.py -------------------------------------------------------------------------------- /compressai/layers/natten/src/natten1dav_cuda.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lumingzzz/TinyLIC/HEAD/compressai/layers/natten/src/natten1dav_cuda.cpp -------------------------------------------------------------------------------- /compressai/layers/natten/src/natten1dav_cuda_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lumingzzz/TinyLIC/HEAD/compressai/layers/natten/src/natten1dav_cuda_kernel.cu -------------------------------------------------------------------------------- /compressai/layers/natten/src/natten1dqkrpb_cuda.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lumingzzz/TinyLIC/HEAD/compressai/layers/natten/src/natten1dqkrpb_cuda.cpp -------------------------------------------------------------------------------- /compressai/layers/natten/src/natten1dqkrpb_cuda_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lumingzzz/TinyLIC/HEAD/compressai/layers/natten/src/natten1dqkrpb_cuda_kernel.cu -------------------------------------------------------------------------------- /compressai/layers/natten/src/natten_commons.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lumingzzz/TinyLIC/HEAD/compressai/layers/natten/src/natten_commons.cuh -------------------------------------------------------------------------------- /compressai/layers/natten/src/nattenav_cuda.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lumingzzz/TinyLIC/HEAD/compressai/layers/natten/src/nattenav_cuda.cpp -------------------------------------------------------------------------------- /compressai/layers/natten/src/nattenav_cuda_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lumingzzz/TinyLIC/HEAD/compressai/layers/natten/src/nattenav_cuda_kernel.cu -------------------------------------------------------------------------------- /compressai/layers/natten/src/nattenqkrpb_cuda.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lumingzzz/TinyLIC/HEAD/compressai/layers/natten/src/nattenqkrpb_cuda.cpp -------------------------------------------------------------------------------- /compressai/layers/natten/src/nattenqkrpb_cuda_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lumingzzz/TinyLIC/HEAD/compressai/layers/natten/src/nattenqkrpb_cuda_kernel.cu -------------------------------------------------------------------------------- /compressai/layers/natten/src/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lumingzzz/TinyLIC/HEAD/compressai/layers/natten/src/setup.py -------------------------------------------------------------------------------- /compressai/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lumingzzz/TinyLIC/HEAD/compressai/models/__init__.py -------------------------------------------------------------------------------- /compressai/models/google.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lumingzzz/TinyLIC/HEAD/compressai/models/google.py -------------------------------------------------------------------------------- /compressai/models/priors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lumingzzz/TinyLIC/HEAD/compressai/models/priors.py -------------------------------------------------------------------------------- /compressai/models/tinylic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lumingzzz/TinyLIC/HEAD/compressai/models/tinylic.py -------------------------------------------------------------------------------- /compressai/models/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lumingzzz/TinyLIC/HEAD/compressai/models/utils.py -------------------------------------------------------------------------------- /compressai/models/video/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lumingzzz/TinyLIC/HEAD/compressai/models/video/__init__.py -------------------------------------------------------------------------------- /compressai/models/video/google.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lumingzzz/TinyLIC/HEAD/compressai/models/video/google.py -------------------------------------------------------------------------------- /compressai/models/waseda.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lumingzzz/TinyLIC/HEAD/compressai/models/waseda.py -------------------------------------------------------------------------------- /compressai/ops/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lumingzzz/TinyLIC/HEAD/compressai/ops/__init__.py -------------------------------------------------------------------------------- /compressai/ops/bound_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lumingzzz/TinyLIC/HEAD/compressai/ops/bound_ops.py -------------------------------------------------------------------------------- /compressai/ops/ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lumingzzz/TinyLIC/HEAD/compressai/ops/ops.py -------------------------------------------------------------------------------- /compressai/ops/parametrizers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lumingzzz/TinyLIC/HEAD/compressai/ops/parametrizers.py -------------------------------------------------------------------------------- /compressai/transforms/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lumingzzz/TinyLIC/HEAD/compressai/transforms/__init__.py -------------------------------------------------------------------------------- /compressai/transforms/functional.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lumingzzz/TinyLIC/HEAD/compressai/transforms/functional.py -------------------------------------------------------------------------------- /compressai/transforms/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lumingzzz/TinyLIC/HEAD/compressai/transforms/transforms.py -------------------------------------------------------------------------------- /compressai/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lumingzzz/TinyLIC/HEAD/compressai/utils/__init__.py -------------------------------------------------------------------------------- /compressai/utils/bench/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lumingzzz/TinyLIC/HEAD/compressai/utils/bench/__init__.py -------------------------------------------------------------------------------- /compressai/utils/bench/__main__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lumingzzz/TinyLIC/HEAD/compressai/utils/bench/__main__.py -------------------------------------------------------------------------------- /compressai/utils/bench/codecs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lumingzzz/TinyLIC/HEAD/compressai/utils/bench/codecs.py -------------------------------------------------------------------------------- /compressai/utils/eval_model/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lumingzzz/TinyLIC/HEAD/compressai/utils/eval_model/__init__.py -------------------------------------------------------------------------------- /compressai/utils/eval_model/__main__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lumingzzz/TinyLIC/HEAD/compressai/utils/eval_model/__main__.py -------------------------------------------------------------------------------- /compressai/utils/find_close/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /compressai/utils/find_close/__main__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lumingzzz/TinyLIC/HEAD/compressai/utils/find_close/__main__.py -------------------------------------------------------------------------------- /compressai/utils/plot/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lumingzzz/TinyLIC/HEAD/compressai/utils/plot/__init__.py -------------------------------------------------------------------------------- /compressai/utils/plot/__main__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lumingzzz/TinyLIC/HEAD/compressai/utils/plot/__main__.py -------------------------------------------------------------------------------- /compressai/utils/update_model/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lumingzzz/TinyLIC/HEAD/compressai/utils/update_model/__init__.py -------------------------------------------------------------------------------- /compressai/utils/update_model/__main__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lumingzzz/TinyLIC/HEAD/compressai/utils/update_model/__main__.py -------------------------------------------------------------------------------- /compressai/utils/video/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lumingzzz/TinyLIC/HEAD/compressai/utils/video/__init__.py -------------------------------------------------------------------------------- /compressai/utils/video/bench/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lumingzzz/TinyLIC/HEAD/compressai/utils/video/bench/__init__.py -------------------------------------------------------------------------------- /compressai/utils/video/bench/__main__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lumingzzz/TinyLIC/HEAD/compressai/utils/video/bench/__main__.py -------------------------------------------------------------------------------- /compressai/utils/video/bench/codecs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lumingzzz/TinyLIC/HEAD/compressai/utils/video/bench/codecs.py -------------------------------------------------------------------------------- /compressai/utils/video/eval_model/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lumingzzz/TinyLIC/HEAD/compressai/utils/video/eval_model/__init__.py -------------------------------------------------------------------------------- /compressai/utils/video/eval_model/__main__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lumingzzz/TinyLIC/HEAD/compressai/utils/video/eval_model/__main__.py -------------------------------------------------------------------------------- /compressai/utils/video/plot/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lumingzzz/TinyLIC/HEAD/compressai/utils/video/plot/__init__.py -------------------------------------------------------------------------------- /compressai/utils/video/plot/__main__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lumingzzz/TinyLIC/HEAD/compressai/utils/video/plot/__main__.py -------------------------------------------------------------------------------- /compressai/zoo/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lumingzzz/TinyLIC/HEAD/compressai/zoo/__init__.py -------------------------------------------------------------------------------- /compressai/zoo/image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lumingzzz/TinyLIC/HEAD/compressai/zoo/image.py -------------------------------------------------------------------------------- /compressai/zoo/pretrained.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lumingzzz/TinyLIC/HEAD/compressai/zoo/pretrained.py -------------------------------------------------------------------------------- /compressai/zoo/video.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lumingzzz/TinyLIC/HEAD/compressai/zoo/video.py -------------------------------------------------------------------------------- /examples/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lumingzzz/TinyLIC/HEAD/examples/train.py -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lumingzzz/TinyLIC/HEAD/pyproject.toml -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lumingzzz/TinyLIC/HEAD/requirements.txt -------------------------------------------------------------------------------- /results/ms-ssim/clic.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lumingzzz/TinyLIC/HEAD/results/ms-ssim/clic.json -------------------------------------------------------------------------------- /results/ms-ssim/kodak.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lumingzzz/TinyLIC/HEAD/results/ms-ssim/kodak.json -------------------------------------------------------------------------------- /results/ms-ssim/tecnick.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lumingzzz/TinyLIC/HEAD/results/ms-ssim/tecnick.json -------------------------------------------------------------------------------- /results/mse/clic.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lumingzzz/TinyLIC/HEAD/results/mse/clic.json -------------------------------------------------------------------------------- /results/mse/kodak.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lumingzzz/TinyLIC/HEAD/results/mse/kodak.json -------------------------------------------------------------------------------- /results/mse/tecnick.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lumingzzz/TinyLIC/HEAD/results/mse/tecnick.json -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lumingzzz/TinyLIC/HEAD/setup.py -------------------------------------------------------------------------------- /third_party/ryg_rans/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lumingzzz/TinyLIC/HEAD/third_party/ryg_rans/LICENSE -------------------------------------------------------------------------------- /third_party/ryg_rans/README: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lumingzzz/TinyLIC/HEAD/third_party/ryg_rans/README -------------------------------------------------------------------------------- /third_party/ryg_rans/rans64.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lumingzzz/TinyLIC/HEAD/third_party/ryg_rans/rans64.h -------------------------------------------------------------------------------- /third_party/ryg_rans/rans_byte.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lumingzzz/TinyLIC/HEAD/third_party/ryg_rans/rans_byte.h -------------------------------------------------------------------------------- /third_party/ryg_rans/rans_word_sse41.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lumingzzz/TinyLIC/HEAD/third_party/ryg_rans/rans_word_sse41.h --------------------------------------------------------------------------------