CompressAI学习

 CompressAI是什么：

CompressAI：InterDigital开源基于学习的图像视频压缩研究库 - 知乎编译 | CV 君 报道 | 我爱计算机视觉公众号（id:aicvml）在多媒体技术应用领域，图像视频编解码居于基础地位，在任何图像和视频应用的存储和传输中，都要涉及图像视频的压缩和解压。在音视频数据不断爆炸式发展的…https://zhuanlan.zhihu.com/p/314961069

调用编码器

python examples/codec.py encode D:VCMCompressAI-masterinputinput.jpg -f 1 --model cheng2020-anchor

调用解码器

python examples/codec.py decode D:VCMCompressAI-masterinput.bin -o D:VCMCompressAI-masteroutout.jpg

 评估预训练编码器的性能

python -m compressai.utils.eval_model pretrained D:VCMCompressAI-masterinput -a cheng2020-anchor -q 2

input -a cheng2020-anchor -q 2
{
  "name": "cheng2020-anchor",
  "description": "Inference (ans)",
  "results": {
    "psnr": [
      29.59088579241053
    ],
    "ms-ssim": [
      0.9556650519371033
    ],
    "bpp": [
      0.19759615384615384
    ],
    "encoding_time": [
      5.068420171737671
    ],
    "decoding_time": [
      9.750147819519043
    ]
  }
}

compressaizooimage 里面有预训练模型的下载地址和对应代码

model_architectures = {
    "bmshj2018-factorized": FactorizedPrior,
    "bmshj2018-hyperprior": ScaleHyperprior,
    "mbt2018-mean": MeanScaleHyperprior,
    "mbt2018": JointAutoregressiveHierarchicalPriors,
    "cheng2020-anchor": Cheng2020Anchor,
    "cheng2020-attn": Cheng2020Attention,
}
其对应代码在compressai.models里面，通过链接可以找到。前4个在google.py里，后两个在waseda.py里