VARIABLE RATE IMAGE COMPRESSION WITH RECURRENT NEURAL NETWORKS代码复现

首先上github地址github地址
这里依旧记录一下复现过程中遇到的问题

1、安装pytorch环境

这一步先跳过了，可以参考之前的文章安装pytorch环境，如果在服务器安装记得选择linux环境

2、安装缺少的包

pip install pyyaml
pip install tb-nightly
pip install colored

3、训练使用绝对路径

github中给出的训练命令是

但实际过程中我调试出现了路径不存在的问题，我将其改为了绝对路径

4、数据集下载

数据集由来自YouTube-8M 数据集的 720p 图像组成，这里csdn不让上传1000M以上的，大家自己下载吧
数据集地址