PyTorch-MSSSIM：快速、可微分的MS-SSIM与SSIM插件
5星
- 浏览量: 0
- 大小:None
- 文件类型：None
简介：
PyTorch-MSSSIM是一款高效计算多尺度结构相似性（MS-SSIM）及结构相似性（SSIM）指标的PyTorch插件，支持自动求导。
PyTorch MS-SSIM 是一个适用于 PyTorch 1.0+ 的快速且可区分的多尺度结构相似性（MS-SSIM）和结构相似度（SSIM）库。更新至2020年8月21日版本 v0.2.1，现在支持3D图像处理；在2020年4月30日发布的 v0.2 版本中，ssim 和 ms-ssim 的计算方法与 TensorFlow 和 scikit-image 相同。基准测试结果可以在库的“测试”部分找到。
安装方式：
```shell
pip install pytorch-msssim
```
使用说明：所有操作均在输入图像所在的设备上进行。
1. 基本用法示例:
```python
from pytorch_msssim import ssim, ms_ssim, SSIM, MS_SSIM
# X: (N,3,H,W) 一批非负RGB图像
```
注意，这里的代码片段为使用库的基本方法提供了简要说明。
全部评论 (0)
还没有任何评论哟~
