Google compress 安装

对于使用深度学习进行图像压缩的同学来说google的compress是绕不过去的一个坎
这里记录一下google compression的安装过程
代码来源
本篇中包含四篇文章的代码

• End-to-end optimized image compression
• Joint autoregressive and hierarchical priors for learned image
  compression
• Channel-wise autoregressive entropy models for learned image compression
• High-Fidelity Generative Image Compression

本次安装的环境：
ubuntu 18.04
GPU：RTX3090
tensorflow：2.5.0

1. 首先建立环境

#conda　create -n tf25 python=3.8
#conda activate tf25

进入到代码目录

#cd dir/compression-master/

2. 安装必要的包

pip install -r requirements.txt
pip install tensorflow-compression

测试安装成功没有

python -m tensorflow_compression.all_tests

• 注意这里可能会出现

tensorflow.python.framework.errors_impl.NotFoundError: /data/tkd/compression-2.2/tensorflow_compression/python/ops/../../cc/libtensorflow_compression.so: cannot open shared object file: No such file or directory

这个时候可以进入到环境目录下

#cd CondaDir/envs/tf25/lib/python3.8/site-packages/tensorflow_compression/cc

将这个文件复制到代码
tensorflow_compression/python/cc/
文件夹下再运行以上程序

其他具体的使用可以参考代码来源中的具体使用教程