图像质量评价(IQA)

1.评价方法

图像质量评价分为:主观和客观两个方面。其中客观又分为全参考、半参考和无参考三种

详情见如下博客内容

https://blog.csdn.net/shinuone/article/details/149176494

2.评价指标和评价算法

GitHub - chaofengc/IQA-PyTorch: 👁️ 🖼️ 🔥PyTorch Toolbox for Image Quality Assessment, including PSNR, SSIM, LPIPS, FID, NIQE, NRQM(Ma), MUSIQ, TOPIQ, NIMA, DBCNN, BRISQUE, PI and more...

IQA-PyTorch/docs/ModelCard.md at main · chaofengc/IQA-PyTorch · GitHubhttps://github.com/chaofengc/IQA-PyTorch/blob/main/docs/ModelCard.md

按照git上的要求搭好环境后使用以下testiqa.py进行测试,图片可以放在绝对路径下

python 复制代码
import pyiqa
# pyiqa是pytorch实现的图像质量评估库,直接pip install
import torch
from PIL import Image
import torchvision.transforms as transforms

# 可以列出左右库里的函数
print(pyiqa.list_models())

device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")

# 示例
iqa_metric = pyiqa.create_metric('arniqa', device=device) #随意选择上述存在的模型,注意是否是NR的
to_tensor = transforms.ToTensor()
img_path=r"E:\code\pytorch_test\IQA-PyTorch-main\IQA-PyTorch-main\test.jpg" #图片路径
img = to_tensor(Image.open(img_path)).unsqueeze(0)
score = iqa_metric(img)
print("NIME:", score.item())

测试结果如下图所示

表示使用该模型arniqa,得到的分数为0.45

相关推荐
hhzz14 小时前
AI数字营销平台会员功能模拟测评:营销组件、数据监测与专家标注实测体验
人工智能
LiuJun2Son14 小时前
Claude Code + Skill 做 UI 的实战工作流
人工智能·ui
wuhen_n14 小时前
RAG 入门:检索增强生成核心原理
前端·人工智能·typescript·langchain·ai编程
继续商行14 小时前
大模型推理优化实战:从KV Cache到流式响应的全链路性能提升
人工智能
冬奇Lab14 小时前
Agent 系列(15):Agent 记忆系统进阶——短期、长期、压缩,三层记忆架构
人工智能·llm·agent
大雨淅淅14 小时前
【机器人】ROS2 机械臂控制(MoveIt2)从入门到实战
人工智能·python·神经网络·学习·算法·机器学习·机器人
m0_5648768414 小时前
怎么写好一个好的skill
人工智能·深度学习·职场和发展
zhangfeng113314 小时前
把权重写死在芯片的架构 Taalas(HC1)芯片:车载 GPU / 智能驾驶 / 机器人 / 算力卡适配总结
人工智能·深度学习·语言模型·架构·机器人·gpu算力·芯片
芝士爱知识a14 小时前
【2026量化新纪元】深度评测:以AlphaGBM为核心的顶级AI量化分析软件推荐及全维度选型指南
人工智能·机器学习·因子挖掘·ai量化·alphagbm·量化交易软件测评
OBiO201314 小时前
精准靶向血管平滑肌AAV在心血管疾病研究中的应用
人工智能