CompressAI从download到run
download
进入CompressAI的Github代码页下载压缩包并解压到自己的项目路径。 path\to\your\compressAI
python environment
CompressAI 需要 PyCharm 进行代码的编辑和 Anaconda 进行 Python 虚拟环境的管理,详细的安装问题和两个软件之间的联动问题在本篇博文中有介绍。 分别下载好这两个软件之后,就可以按照教程中的指引去在 Anaconda 下安装 PyTorch 的环境。
Project environment
准备好这一切之后就可以开始尝试跑起来 CompressAI 了,再次回到 Github代码页,进入 readme,按照 readme 所示流程进行 CompressAI 的安装和测试。
在已经下载压缩包的前提下,我们只需要将压缩包解压,然后在解压文件夹path\to\your\compressAI
下运行如下命令(注意此文件夹内有setup.py文件):
pip install -U pip
pip install -e .
该命令应当运行在之前创建的 PyTorch 虚拟环境下,此时就将 CompressAI 的根目录安装到虚拟环境下了。
requirements
经过环境的创建以及项目的安装之后,我们就到了 run 之前的最后一步:配置依赖
同样的,进入项目文件夹path\to\your\compressAI
后,在之前创建的虚环境下运行:
conda:
conda install --file requirements.txt
pip:
pip install -r requirements.txt
运行过程中可能出现某些包在 conda 源之下不可达的问题,对于这些包先用pip进行安装,随后再用conda安装剩余的包即可
run
在完成了上述所有操作之后,恭喜你,你可以开始运行 CompressAI 了。
使用 PyCharm 打开项目文件夹,在 Terminal 中 cd 进入 examples 路径下,输入 python codec.py --help
测试是否安装成功,如果出现以下结果,即为安装成功;如果出现其他结果,那就是哪里出了问题,根据报错去排查即可: