CompresssAI介绍
CompressAI 是将四篇基于深度学习端到端图像压缩代码从tensorflow搬移到了pytorch上,提供了完整的实例代码和使用教程,具体可以看CompressAI的Github官方库,同时提供了与传统图像编码方式的对比。CompressAI对图像压缩领域的新手来说是一个比较好的入手方向。
项目地址:CompressAI
项目使用教程
四篇端到端图像编码算法代码:
《End-to-end Optimized Image Compression》
《Variational Image Compression With A Scale Hyperprior》
《Joint Autoregressive and Hierarchical Priors for Learned Image Compression》
《Learned Image Compression with Discretized Gaussian Mixture Likelihoods and
Attention Modules》
环境安装
使用conda环境安装pytorch 1.7.1 cuda=11.0
conda create -n env_name python=3.8
安装pytorch
pip install torch==1.7.1+cu110