tensorflow-compression项目部署记录
github链接: https://github.com/tensorflow/compression.
最近看了这篇端到端的文章,想尝试一下部署运行,复现一下,结果遇到很多问题,先就此记录着。
初次部署问题记录1(已解决)
-
预编译包目前仅针对 Linux 和 Darwin/Mac OS 提供。要在 Windows 上使用这些包,必须需要先下载docker,然后在docker中使用代码:pip install tensorFlow-compression 才能下载的下来(anaconda、直接的pip都不行);
-
Docker下载tensorflow映像,
链接: https://www.tensorflow.org/install/docker.使用代码:docker run -it --rm tensorflow/tensorflow每次打开的docker容器id是不同的,之前在某一个印象中下载了tensorflow-compression,后来再进去发现运行不起来,才知道每次就算名字一样进入的是不同的容器,环境不一样; -
关于tfci.py文件:module没找到是包没导入好,环境中要有tensorflow、tfc;目前成功的只有docker内pip下载方法,运行格式:
python tfci.py compress < model > < PNG file >,其中model可选项为:“hific-lo”, “hific-mi”, “hific-hi”, 但是如何在docker中运行这个文件有待考究; -
官网提供的colab内可以实现自己的图片交互式运行体验: https://colab.research.google.com/github/tensorflow/compression/blob/master/models/hific/colab.ipynb.;
-
该giithub下另一个readme文件https://github.com/tensorflow/compression/tree/master/models/hific.使用git下载下来之后运行提示缺少tensorflow-compression工具,要下载又回到上面的问题了。
之后可以进行的工作
- docker稍微学习一下docker官方文档,看看是否可以使用docker环境运行这个python代码;
- colab代码看看,用用体验一下;
- 让pycharm利用docker的环境进行开发 参考csdn链接或pycharm+docker,感觉有点复杂,不知道方向对不对,先搁置。
- 虚拟机配置合适环境。
ps:哎,这个项目在Windows下复现真的有点不友好o(╥﹏╥)o
再次虚拟机部署更新
既然官网说只能运行在maxos或者linux上,那么就可以部署一台linux虚拟机,然后再往下进行。
步骤
- 下载VMware虚拟机和centos镜像,按照教程步骤配置。
- 在虚拟机中安装anaconda后,进入目录通过bash命令安装。
- 按照github配置环境,下载tensorflow_compression(中间可能会断几次,毕竟下载的确实有点慢,多来几次就好了)。
- 配置中输入python,输入 import tensorflow_compression as tf 不报错即可继续进行。
操作
a pre-trained model,model可选"hific-lo", “hific-mi”, “hific-hi”,当然 python tfci.py models 可以列出所有的model可选项。
python tfci.py compress <model> <PNG file>
实际操作
cd /home/karen/Downloads
conda activate ENV_NAME
python tfci.py compress hific-lo tulip.png
问题记录2(有待解决)
运行上述代码后遇到这个问题,明天查一下看看能不能解决。
PNG warning: iCCP: known incorrect sRGB profile 这警告可以忽略,非要处理可以参照这篇博客.
其他几处错误:或多或少和GPU处理有关,需要用到有GPU的电脑。