前言
文献名称:Learned Image Compression with Discretized Gaussian Mixture Likelihoods and Attention Modules
本文基于CompressAI的库进行复现
github地址:compressAI
关于compressAI相关博客说明:CompressAI:基于pytorch的图像压缩框架使用
安装好compressai后相当于把这个底层库引入了我们的工程
相关环境搭配可以参考Joint Autoregressive and Hierarchical Priors for Learned Image Compression文献复现文献复现
同样都是使用compressAI进行复现
1、创建工程
我这里在做好compressAI所需前置工作后,把我需要的几个文件单独移入了自己的工程
2、创建训练命令脚本
因为这篇文献其实是可以直接调用相关命令对实验进行复现,所以主要把相关命令输入,按照步骤一步步执行即可,相关脚本命令