Advertisement

MS-SSIM-PyTorch:基于PyTorch JIT的高效SSIM与MS-SSIM实现代码

  •  5星
  •     浏览量: 0
  •     大小:None
  •      文件类型:ZIP


简介:
MS-SSIM-PyTorch是一个利用PyTorch JIT技术优化过的高效计算结构相似性指标(SSIM)及其多尺度版本(MS-SSIM)的代码库,适用于图像和视频的质量评估。 这段代码是从已有的代码基础上修改而来的,并且部分代码已经经过优化以提高速度、减少VRAM占用并兼容pytorch jit。 还有一个动态频道版本可供选择,这个版本使用起来更加方便,但性能损失很小。 感谢贡献者找到并修复了ms_ssim在反向传播时会导致梯度nan的问题。 如果您正在使用的是pytorch 1.2,请注意不要在训练循环中创建和销毁该jit模块(其他jit模块也可能出现这种情况),这可能会导致内存泄漏。我已经测试过,在pytorch 1.6版本上没有这个问题。 我还研究了piqa库,这使得我实现的ssim和ms-ssim的速度比以前更快了一些。 加速效果仅在GPU上进行了验证。 losser1是268fc76, losser2是881d210, losser3是5caf547, losser4是1c2f14a, loser5是abaf398,其中最新的版本为abaf398。 在pytorch 1.7中进行了相应的优化。

全部评论 (0)

还没有任何评论哟~
客服
客服
  • MS-SSIM-PyTorchPyTorch JITSSIMMS-SSIM
    优质
    MS-SSIM-PyTorch是一个利用PyTorch JIT技术优化过的高效计算结构相似性指标(SSIM)及其多尺度版本(MS-SSIM)的代码库,适用于图像和视频的质量评估。 这段代码是从已有的代码基础上修改而来的,并且部分代码已经经过优化以提高速度、减少VRAM占用并兼容pytorch jit。 还有一个动态频道版本可供选择,这个版本使用起来更加方便,但性能损失很小。 感谢贡献者找到并修复了ms_ssim在反向传播时会导致梯度nan的问题。 如果您正在使用的是pytorch 1.2,请注意不要在训练循环中创建和销毁该jit模块(其他jit模块也可能出现这种情况),这可能会导致内存泄漏。我已经测试过,在pytorch 1.6版本上没有这个问题。 我还研究了piqa库,这使得我实现的ssim和ms-ssim的速度比以前更快了一些。 加速效果仅在GPU上进行了验证。 losser1是268fc76, losser2是881d210, losser3是5caf547, losser4是1c2f14a, loser5是abaf398,其中最新的版本为abaf398。 在pytorch 1.7中进行了相应的优化。
  • PyTorch-MSSSIM:快速、可微分MS-SSIMSSIM插件
    优质
    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图像 ``` 注意,这里的代码片段为使用库的基本方法提供了简要说明。
  • MS-SSIM 图像质量评估
    优质
    本研究提出了一种基于MS-SSIM(多尺度结构相似性指数)的图像质量客观评价方法,旨在更准确地衡量图像处理后的视觉效果变化。该方法综合考虑了不同尺度下的图像特征,能够有效捕捉人眼对图像细节和结构敏感的特点,为图像质量评估提供了一个更为可靠的量化标准。 图像质量评价全参考MSSSIM算法可以直接运行。
  • Matlab灰度处理Python IW-SSIMPyTorch
    优质
    本项目介绍如何使用Matlab进行图像灰度处理,并用Python结合PyTorch框架实现IW-SSIM算法,提供源码和详细注释。 MATLAB灰度处理代码可以用于将彩色图像转换为灰度图。这种处理在计算机视觉和图像处理领域非常常见,因为它简化了图像数据并减少了计算复杂性。实现这一功能的常用方法是使用内置函数`rgb2gray()`,该函数根据人眼对不同颜色敏感程度的不同来调整RGB值以生成单通道灰度图像。 另外一种方式是从头开始编写代码进行转换,这需要理解如何通过加权平均R、G和B分量计算每个像素的亮度。这种方法允许用户自定义权重,从而适应不同的视觉需求或特定的应用场景。 在处理过程中可能还需要对得到的灰度图进一步操作,比如调整对比度或者应用滤波器来增强细节或是去除噪声等。这些步骤都是通过一系列MATLAB内置函数实现的,并且可以根据具体的需求灵活组合使用。
  • MATLABSSIM
    优质
    本文章介绍了如何使用MATLAB软件实现结构相似性指数(SSIM)算法,并探讨其在图像质量评估中的应用。 经典full-reference图像质量评估标准SSIM的MATLAB代码实现。
  • PSNRSSIM
    优质
    这段代码提供了计算图像处理中常用的两个评价指标——峰值信噪比(PSNR)和结构相似性指数(SSIM)的方法。适用于评估图像压缩、去噪等任务的效果。 C++和OpenCV编写的PSNR+SSIM代码用于测试图像质量。
  • SSIMMATLAB
    优质
    这段MATLAB代码实现了结构相似性指数(SSIM)算法,用于评估两幅图像之间的视觉质量相似度,适用于图像处理和计算机视觉领域的研究与开发。 这是一段可使用的SSIM代码,只需更改图片的位置即可。
  • PyTorch超分辨率重建PSNRSSIM计算包.zip
    优质
    本资源提供一个利用PyTorch实现图像超分辨率重建效果评估的代码包,包含PSNR和SSIM两个关键指标的计算方法,便于研究者测试模型性能。 整理的超分重建 SR 和 HR 图片 PSNR 和 SSIM 计算(pytorch 实现)代码可以在这里找到;如果珍惜积分的话,可以在 https://positive.blog..net/article/details/109688295 博文手动粘贴,内容是一样的。 去掉链接后: 整理的超分重建 SR 和 HR 图片 PSNR 和 SSIM 计算(pytorch 实现)代码可以在这里找到;如果珍惜积分的话,可以在相关博文手动粘贴,内容是一样的。
  • MATLABSSIM算法
    优质
    本项目基于MATLAB平台实现了结构相似性(SSIM)图像质量评估算法,可用于评价处理后图像与原图之间的视觉效果差异。 SSIM(Structural Similarity Index,结构相似性指标)是一种衡量图像质量的度量方法,由Wang等人在2004年提出。它比传统的均方误差(MSE)和峰值信噪比(PSNR)更能反映出人眼对图像质量的感知。在MATLAB中实现SSIM算法,我们可以直接调用MATLAB提供的`ssim`函数,或者自定义代码来完成。这里我们将深入探讨SSIM算法的原理、MATLAB实现以及如何使用。 **SSIM算法原理** SSIM算法主要通过比较两幅图像的亮度、对比度和结构三个方面的相似性来评估它们的质量。计算公式为: \[ SSIM(x,y) = \frac{(2mu_xmu_y + c_1)(2sigma_{xy} + c_2)}{(mu_x^2 + mu_y^2 + c_1)(sigma_x^2 + sigma_y^2 + c_2)} \] 其中,\( x \) 和 \( y \) 分别代表两幅图像的像素值,\( mu_x \) 和 \( mu_y \) 是它们的平均值,\( sigma_x^2 \) 和 \( sigma_y^2 \) 是它们的方差,\( sigma_{xy} \) 是它们的协方差,\( c_1 \) 和 \( c_2 \) 是常数,通常取值为 \( (k_1L)^2 \) 和 \( (k_2L)^2 \),\( L \) 是图像的最大灰度值,\( k_1 \) 和 \( k_2 \) 通常取值为0.01和0.03。 **MATLAB实现** MATLAB提供了内置函数`ssim`来计算SSIM值。以下是一个简单的使用示例: ```matlab % 加载两幅图像 img1 = imread(image1.jpg); img2 = imread(image2.jpg); % 调整图像大小,确保两幅图具有相同的尺寸 if size(img1,1) ~= size(img2,1) || size(img1,2) ~= size(img2,2) img1 = imresize(img1, size(img2)); end % 计算SSIM值 ssimValue = ssim(img1, img2); ``` **自定义SSIM实现** 如果你需要自定义SSIM算法,可以按照以下步骤进行: 1. **计算均值和方差**:对于每幅图像,计算每个窗口(如8x8像素)内的均值和方差。 2. **计算协方差**:在相同窗口内,计算两幅图像对应像素的协方差。 3. **归一化**:使用常数 \( c_1 \) 和 \( c_2 \) 来避免除以零的情况,同时确保SSIM值在-1到1之间。 4. **计算SSIM分块值**:使用上述公式计算每个窗口的SSIM值。 5. **求平均**:将所有窗口的SSIM值平均得到整个图像的SSIM值。 例如,自定义的MATLAB代码可能如下: ```matlab function [ssimValue] = custom_ssim(img1, img2) % 参数设置 windowSize = 8; k1 = 0.01; k2 = 0.03; L = max(max(img1), max(img2)); % 计算均值、方差和协方差 mu1 = movmean(img1, windowSize); mu2 = movmean(img2, windowSize); sigma12 = movcov(img1, img2, windowSize); sigma1 = movvar(img1, windowSize); sigma2 = movvar(img2, windowSize); % 归一化 c1 = (k1*L)^2; c2 = (k2*L)^2; % 计算SSIM分块值 ssimBlock = (2*mu1.*mu2 + c1)*(2*sigma12 + c2)./(mu1.^2 + mu2.^2 + c1).*(sigma1 + sigma2 + c2).^(-1); % 求平均 ssimValue = mean(mean(ssimBlock)); end ``` 以上是SSIM算法的基本介绍及其在MATLAB中的实现方式。在实际应用中,SSIM常用于图像压缩、图像增强、视频编码等领域的图像质量评估。通过对比SSIM值,我们可以判断图像处理过程是否导致了图像质量的显著下降。由于SSIM考虑了图像的结构信息,因此在评价人眼视觉感知的图像质量时,SSIM比传统的MSE和PSNR更具有优势。
  • Python中SSIM
    优质
    本文介绍了如何在Python编程语言中实现结构相似性指数(SSIM)算法,探讨了其在图像质量评估中的应用。 Python实现的图像结构相似性度量(Structural Similarity Image Metric, SSIM)。