├── README.md ├── _config.yml ├── code ├── Model │ ├── GDN_transform.py │ ├── basic_module.py │ ├── context_model.py │ ├── factorized_entropy_model.py │ ├── gaussian_entropy_model.py │ └── model.py ├── README.md ├── Util │ ├── AE │ │ ├── AE.cpp │ │ ├── My_Range_Coder.h │ │ ├── My_Range_Decoder.cpp │ │ ├── My_Range_Encoder.cpp │ │ ├── README.md │ │ └── setup.py │ ├── Dockerfile │ ├── metrics.py │ ├── msssim.py │ └── torch_msssim.py ├── Weights │ └── README.md ├── inference.py ├── inference_rd_fast.py ├── select_rd_results.py ├── train.py └── train_msssim.py └── images ├── Tecnick_MSSSIM.png ├── Tecnick_PSNR.png └── visual_com.png /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NJUVISION/NIC/HEAD/README.md -------------------------------------------------------------------------------- /_config.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NJUVISION/NIC/HEAD/_config.yml -------------------------------------------------------------------------------- /code/Model/GDN_transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NJUVISION/NIC/HEAD/code/Model/GDN_transform.py -------------------------------------------------------------------------------- /code/Model/basic_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NJUVISION/NIC/HEAD/code/Model/basic_module.py -------------------------------------------------------------------------------- /code/Model/context_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NJUVISION/NIC/HEAD/code/Model/context_model.py -------------------------------------------------------------------------------- /code/Model/factorized_entropy_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NJUVISION/NIC/HEAD/code/Model/factorized_entropy_model.py -------------------------------------------------------------------------------- /code/Model/gaussian_entropy_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NJUVISION/NIC/HEAD/code/Model/gaussian_entropy_model.py -------------------------------------------------------------------------------- /code/Model/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NJUVISION/NIC/HEAD/code/Model/model.py -------------------------------------------------------------------------------- /code/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NJUVISION/NIC/HEAD/code/README.md -------------------------------------------------------------------------------- /code/Util/AE/AE.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NJUVISION/NIC/HEAD/code/Util/AE/AE.cpp -------------------------------------------------------------------------------- /code/Util/AE/My_Range_Coder.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NJUVISION/NIC/HEAD/code/Util/AE/My_Range_Coder.h -------------------------------------------------------------------------------- /code/Util/AE/My_Range_Decoder.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NJUVISION/NIC/HEAD/code/Util/AE/My_Range_Decoder.cpp -------------------------------------------------------------------------------- /code/Util/AE/My_Range_Encoder.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NJUVISION/NIC/HEAD/code/Util/AE/My_Range_Encoder.cpp -------------------------------------------------------------------------------- /code/Util/AE/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NJUVISION/NIC/HEAD/code/Util/AE/README.md -------------------------------------------------------------------------------- /code/Util/AE/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NJUVISION/NIC/HEAD/code/Util/AE/setup.py -------------------------------------------------------------------------------- /code/Util/Dockerfile: -------------------------------------------------------------------------------- 1 | FROM tongxyh/pytorch1.3_ae:latest 2 | -------------------------------------------------------------------------------- /code/Util/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NJUVISION/NIC/HEAD/code/Util/metrics.py -------------------------------------------------------------------------------- /code/Util/msssim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NJUVISION/NIC/HEAD/code/Util/msssim.py -------------------------------------------------------------------------------- /code/Util/torch_msssim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NJUVISION/NIC/HEAD/code/Util/torch_msssim.py -------------------------------------------------------------------------------- /code/Weights/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NJUVISION/NIC/HEAD/code/Weights/README.md -------------------------------------------------------------------------------- /code/inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NJUVISION/NIC/HEAD/code/inference.py -------------------------------------------------------------------------------- /code/inference_rd_fast.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NJUVISION/NIC/HEAD/code/inference_rd_fast.py -------------------------------------------------------------------------------- /code/select_rd_results.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NJUVISION/NIC/HEAD/code/select_rd_results.py -------------------------------------------------------------------------------- /code/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NJUVISION/NIC/HEAD/code/train.py -------------------------------------------------------------------------------- /code/train_msssim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NJUVISION/NIC/HEAD/code/train_msssim.py -------------------------------------------------------------------------------- /images/Tecnick_MSSSIM.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NJUVISION/NIC/HEAD/images/Tecnick_MSSSIM.png -------------------------------------------------------------------------------- /images/Tecnick_PSNR.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NJUVISION/NIC/HEAD/images/Tecnick_PSNR.png -------------------------------------------------------------------------------- /images/visual_com.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NJUVISION/NIC/HEAD/images/visual_com.png --------------------------------------------------------------------------------