文件名称:CompressAI:用于端到端压缩研究的PyTorch库和评估平台
文件大小:7.29MB
文件格式:ZIP
更新时间:2024-04-09 02:06:01
python machine-learning deep-neural-networks compression deep-learning
压缩AI CompressAI( compress-ay )是用于端到端压缩研究的PyTorch库和评估平台。 CompressAI当前提供: 用于基于深度学习的数据压缩的自定义操作,层和模型 官方库的部分端口 预训练的端到端压缩模型,用于学习图像压缩 评估脚本,将学习的模型与经典图像/视频压缩编解码器进行比较 注意:多GPU支持目前处于试验阶段。 安装 CompressAI仅支持python 3.6+(当前对PyTorch的支持<3.9)和PyTorch 1.4+。还需要C ++ 17编译器,最新版本的pip(19.0+)和常见的python软件包(有关完整列表,请参见setup.py )。 要开始并安装CompressAI,请在运行以下命令: git clone https://github.com/InterDigitalInc/CompressAI compressai cd