文件名称：ms_ssim_pytorch:使用pytorch jit的快速ssim和ms-ssim实现代码
文件大小：23.39MB
文件格式：ZIP
更新时间：2024-06-17 10:01:50
Python
ms_ssim_pytorch 该代码是从修改的。 部分代码已被修改以使其更快，占用更少的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 losser5是 abaf398 abaf398 在pytorch 1.7.
【文件预览】：
ms_ssim_pytorch-master
----ms_ssim_test.mkv(4.42MB)
----.gitattributes(66B)
----make_gif.cmd(131B)
----ssim.py(12KB)
----ms_ssim_test.gif(10.7MB)
----LICENSE(1KB)
----README.md(6KB)
----ssim_test.gif(5.85MB)
----ssim_test.mkv(2.34MB)
----.gitignore(1KB)
----_test_speed()
--------no1_ms_ssim_lizhengwei1992_MS_SSIM_pytorch.py(3KB)
--------test_ssim_speed.py(2KB)
--------test_ms_ssim_speed.py(2KB)
--------no3_ssim_VainF_pytorch_msssim.py(11KB)
--------no2_ssim_Po_Hsun_Su_pytorch_ssim.py(3KB)
--------no5_ssim_francois_rozet_piqa.py(19KB)
----test_img1.jpg(120KB)