PyTorch中的IQA模型
IQA-pytorch的Python项目详细描述
Pythorch图像质量评价模型
这是一个使用PyTorch重新实现现有IQA模型的存储库，包括
注意：重现的结果可能与最初的matlab版本略有不同。在
安装：
pip install IQA_pytorch
要求：
- Python>；=3.6
- Pythorch>；=1.2
用法：
fromIQA_pytorchimportSSIM,GMSD,LPIPSvgg,DISTSD=SSIM()# Calculate score of the image X with the reference Y# X: (N,3,H,W) # Y: (N,3,H,W) # Tensor, data range: 0~1score=D(X,Y,as_loss=False)# set 'as_loss=True' to get a value as loss for optimizations.loss=D(X,Y)loss.backward()
- 项目
标签：
欢迎加入QQ群-->： 979659372
