更快的 SSIM & MS-SSIM pytorch 实现 和 收敛过程GIF

训练自编码机，好闲啊。。。就随便优化了一下github上弄下来的 SSIM 类。发现性能有极大的提升！好，专心一下优化这东西。

更快的 SSIM & MS-SSIM 的 github 仓库
https://github.com/One-sixth/ms_ssim_pytorch

SSIM 收敛过程 和 MS-SSIM 的收敛过程，图有点大，所以存在 github 的仓库上，也许要翻墙。
仓库内有下面两图的原始mkv视频文件，下载下来用播放器看更流畅。
试试上传B站会怎么样233。
https://www.bilibili.com/video/av56634788/


优化的效果不错，相比最初始的实现加速接近3倍
相比另外一个大佬优化的实现，我是根据这个大佬的实现进行修改的，加速差不多1.5倍
都是 pytorch 更新的好处，现在可以写更快的代码了。

Performance Testing SSIM

testing losser2
cuda time 89041.5390625
perf_counter time 86.8486872

testing losser3
cuda time 51062.2109375
perf_counter time 50.9808635

testing losser4
cuda time 34861.1484375
perf_counter time 34.066023599999994

Performance Testing MS_SSIM

testing losser1
cuda time 140149.578125
perf_counter time 140.040425

testing losser3
cuda time 92967.7890625
perf_counter time 92.9019308

testing losser4
cuda time 55630.640625
perf_counter time 55.594013200000006