CompressAI:基于pytorch的图像压缩框架使用

前言

CompressAI: a PyTorch library and evaluation platform forend-to-end compression research,我的理解是一个基于图像(视频)压缩的API库。他是建立在 PyTorch 之上的,用于基于深度学习的数据压缩的自定义操作、层和模型,其中包括了基于tensorflow.compression压缩包的部分移植,还包括包括一些用于压缩任务的预训练模型。其中可以直接用到的模型来自以下几篇文献:
在这里插入图片描述

  1. bmshj2018-factorized
    Ballé J, Laparra V, Simoncelli E P. End-to-end optimized image compression[C]//ICLR 2017
  2. bmshj2018-hyperprior
    Ballé J, Minnen D, Singh S, et al. Variational image compression with a scale hyperprior[C]//ICLR2018
  3. mbt2018-mean
    Ballé J, Toderici G. Joint autoregressive and hierarchical priors for learned image compression[C]//NIPS 2018.
  4. mbt2018
    Minnen D, Ballé J, Toderici G. Joint autoregressive and hierarchical priors for learned image compression[C]//NIPS 2018.
  5. cheng2020-anchor
    Cheng Z

猜你喜欢

转载自blog.csdn.net/m0_47146037/article/details/125745646