是否有SSIM或甚至MS-SSIM实现TensorFlow?
SSIM(结构相似度指数metric)是一种度量图像质量或图像相似度的指标。它受到人类感知的启发,根据几篇论文,它是一个比l1/l2好得多的损失函数。例如,请参见Loss Functions for Neural Networks for Image Processing。
到目前为止,我在TensorFlow中找不到实现。在试图从C++或Python代码(如Github: VQMT/SSIM)中移植它之后,我就陷入了像在TensorFlow中使用高斯模糊的方法。
有没有人已经试着自己去实现它?
这似乎是你要找的:
msssim.py
用法:
现在有点晚了,但是新版本的TensorFlow(目前是1.9,1.10)有一个内置的功能。检查这里:TensorFlow MS-SSIM。
你需要在会话中运行它。
在深入研究了其他一些python实现之后,我终于可以在TensorFlow中实现一个运行示例:
下面是如何运行它:
如果您发现错误,请告诉我:)
编辑: 此实现仅支持灰度图像
相关问题 更多 >
编程相关推荐