文献名称:Joint Autoregressive and Hierarchical Priors for Learned Image Compression
本文基于CompressAI的库进行复现
github地址:compressAI
关于compressAI相关博客说明:CompressAI:基于pytorch的图像压缩框架使用
我这里在做好compressAI所需前置工作后,把我需要的几个文件单独移入了自己的工程
因为这篇文献其实是可以直接调用相关命令对实验进行复现,所以主要把相关命令输入,按照步骤一步步执行即可,相关脚本命令可以参考sh脚本运行命令
创建成功后:
conda activate PTGPUPY37 python3 train.py -m mbt2018 -d /home/ll/code/CompressAI_reappearance/imageNet --epochs 500 -lr 1e-4 --lambda 0.0016 --batch-size 16 --cuda --save> log.out
其中patchsize取的默认256
具体的命令说明可以参见官网以及CompressAI:基于pytorch的图像压缩框架使用



