Project Icon

IQA-PyTorch

纯Python和PyTorch图像质量评估工具箱

IQA-PyTorch是一款基于纯Python和PyTorch的图像质量评估工具箱,支持多种主流全参考和无参考评估指标。通过GPU加速,评估速度优于Matlab实现,用户可通过命令行或代码进行图像质量评估。该工具箱还支持作为损失函数使用,提供便捷的基准数据集下载和详细文档,适用于评估各种场景。定期更新及多种预训练模型让它成为图像质量评估的理想选择。详情请查阅文档和示例代码。

IQA-PyTorch 项目介绍

简介

IQA-PyTorch 是一个使用纯 Python 和 PyTorch 构建的全面图像质量评估工具箱。该工具箱实现了多个主流的全参考(FR)和无参考(NR)指标,通过 GPU 加速,性能大多超过了传统的 Matlab 实现。项目设计旨在为研究人员和开发人员提供灵活高效的图像质量评估工具。

项目更新

  • 2024年11月:新增 pyiqa.load_dataset 功能,简化常见数据集的加载。
  • 2024年10月:更新 topiq_nr-face 指标,采用 GCFIQA 数据集进行训练。
  • 2024年8月:新增多个指标,如 qalign_4bitqalign_8bit,优化内存使用。
  • 更多更新包括新指标的加入和现有指标的优化。

快速开始

安装

用户可以通过以下方法安装 IQA-PyTorch:

pip install pyiqa  # 使用 pip 安装
pip install git+https://github.com/chaofengc/IQA-PyTorch.git  # 安装最新的 GitHub 版本

基本使用

用户可以通过命令行界面简单地使用此工具包:

pyiqa -ls  # 列出所有可用的指标
pyiqa [metric_name(s)] -t [image_path or dir] -r [image_path or dir] --device [cuda or cpu] --verbose  # 使用默认设置进行测试

进阶功能

用户可以通过代码进行更高级的测试和训练。例如:

import pyiqa
device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
iqa_metric = pyiqa.create_metric('lpips', device=device)
score_fr = iqa_metric('./ResultsCalibra/dist_dir/I03.bmp', './ResultsCalibra/ref_dir/I03.bmp')

用户还可以将指标作为损失函数使用,或加载自定义设置和权重以进行模型重训或微调。

数据集支持

IQA-PyTorch 提供了方便的方式来加载流行的图像质量评估数据集,这些数据集可以通过配置文件自动从 Huggingface 平台下载。

from pyiqa import get_dataset_info, load_dataset
dataset = load_dataset('koniq10k', data_root='./datasets')

基准性能评估

项目附带了多种评估协议,以确保结果的准确性和再现性。用户可以通过提供的脚本获取不同数据集上的性能基准。

参与和贡献

任何对该项目的贡献都是受欢迎的。感兴趣的开发者可以参考贡献指南进行参与。

许可证与引用

本项目遵循 NTU S-Lab 许可协议。用户在使用本项目时请注明来源,并在相关研究中引用相关文献。

联系

项目相关问题可以通过邮箱 chaofenghust@gmail.com 联系。

IQA-PyTorch 提供了丰富的工具和资源来支持图像质量评估的研究和开发,期待您的使用和反馈。

项目侧边栏1项目侧边栏2
推荐项目
Project Cover

豆包MarsCode

豆包 MarsCode 是一款革命性的编程助手,通过AI技术提供代码补全、单测生成、代码解释和智能问答等功能,支持100+编程语言,与主流编辑器无缝集成,显著提升开发效率和代码质量。

Project Cover

AI写歌

Suno AI是一个革命性的AI音乐创作平台,能在短短30秒内帮助用户创作出一首完整的歌曲。无论是寻找创作灵感还是需要快速制作音乐,Suno AI都是音乐爱好者和专业人士的理想选择。

Project Cover

白日梦AI

白日梦AI提供专注于AI视频生成的多样化功能,包括文生视频、动态画面和形象生成等,帮助用户快速上手,创造专业级内容。

Project Cover

有言AI

有言平台提供一站式AIGC视频创作解决方案,通过智能技术简化视频制作流程。无论是企业宣传还是个人分享,有言都能帮助用户快速、轻松地制作出专业级别的视频内容。

Project Cover

Kimi

Kimi AI助手提供多语言对话支持,能够阅读和理解用户上传的文件内容,解析网页信息,并结合搜索结果为用户提供详尽的答案。无论是日常咨询还是专业问题,Kimi都能以友好、专业的方式提供帮助。

Project Cover

讯飞绘镜

讯飞绘镜是一个支持从创意到完整视频创作的智能平台,用户可以快速生成视频素材并创作独特的音乐视频和故事。平台提供多样化的主题和精选作品,帮助用户探索创意灵感。

Project Cover

讯飞文书

讯飞文书依托讯飞星火大模型,为文书写作者提供从素材筹备到稿件撰写及审稿的全程支持。通过录音智记和以稿写稿等功能,满足事务性工作的高频需求,帮助撰稿人节省精力,提高效率,优化工作与生活。

Project Cover

阿里绘蛙

绘蛙是阿里巴巴集团推出的革命性AI电商营销平台。利用尖端人工智能技术,为商家提供一键生成商品图和营销文案的服务,显著提升内容创作效率和营销效果。适用于淘宝、天猫等电商平台,让商品第一时间被种草。

Project Cover

AIWritePaper论文写作

AIWritePaper论文写作是一站式AI论文写作辅助工具,简化了选题、文献检索至论文撰写的整个过程。通过简单设定,平台可快速生成高质量论文大纲和全文,配合图表、参考文献等一应俱全,同时提供开题报告和答辩PPT等增值服务,保障数据安全,有效提升写作效率和论文质量。

投诉举报邮箱: service@vectorlightyear.com
@2024 懂AI·鲁ICP备2024100362号-6·鲁公网安备37021002001498号