Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

如果我想在Ychannel测试图像指标,应该输入怎样的命令? #68

Closed
automatic0313 opened this issue Mar 29, 2023 · 1 comment

Comments

@automatic0313
Copy link

我使用 python inference_iqa.py -m NRQM -i img_root,貌似不能满足我的需求

@chaofengc
Copy link
Owner

测试demo文件并未提供具体方法的option,你需要按照使用样例中的例子写代码测试:

import pyiqa
import torch

iqa_metric = pyiqa.create_metric('nrqm', test_y_channel=True, color_space='ycbcr').to(device)

# Tensor inputs, img_tensor: (N, 3, H, W), RGB, 0 ~ 1
score_nr = iqa_metric(img_tensor_x)

请务必注意使用正确的colorspace。YCBCR和普通的gray channel (对应YIQ) 有所不同。

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants