Compressai模型复现笔记

在学习智能图像压缩中不可避免的会遇到老师要求复现compressAI的情况。这是我的记录笔记。

1.首先是在官网地址下载 

https://github.com/InterDigitalInc/CompressAI compressai

2.将它保存在自己的文件夹里面，（我自己用的是压缩包解压的形式，所以放到新的文件夹里面还要解压）

path\to\you\compressai 

 3.在vscode打开这个compressai 的文件夹之后就是重点了——配置虚拟环境

pip install -U pip

pip install -e . -i https://pypi.tuna.tsinghua.edu.cn/simple 

 如果第二条指令报错的话就可能要加上虚拟环境python的地址。

接着还要配置上compressai的库，套上清华镜像源下载快一些。

pip install compressai -i https://pypi.tuna.tsinghua.edu.cn/simple

如果compressai怎么跑都跑不动，报各种各样的错80%的可能性是环境有问题。我自己就配过好几遍环境，要着重注意一下python,numpy,torch,torchvision等这几个库的匹配问题，它们极有可能出现版本不匹配从而跑不动的问题。

还有就是上面这几个流程走完之后如果想要验证一下是不是环境配置好了可以在终端激活环境之后运行一下codec.py这个文件。（我的虚拟环境名字就叫compressai)

conda activate compressai

cd examples

python codec.py&n