如何在keras中实现损失函数的ssim？
如何在keras中实现损失函数的ssim？
提问于 2019-04-10 03:37:11
我需要SSIM作为我的网络中的一个损失函数，但我的网络有两个输出。我需要使用SSIM进行第一次输出，使用cross-entropy进行下一次输出。损失函数是它们的组合。然而，我需要更高的SSIM和更低的cross-entropy，所以我认为它们的组合是不正确的。另一个问题是我在keras中找不到SSIM的实现。
Tensorflow有tf.image.ssim，但它接受图像，我不认为我可以在损失函数中使用它，对吧？你能告诉我我该怎么做吗？我是keras和深度学习的初学者，我不知道如何将SSIM作为keras中的自定义损失函数。
回答 1
Stack Overflow用户
发布于 2019-04-10 03:58:15
看看Keras中SSIM loss的here。您可以编写自定义损失函数，并为一个预测创建SSIM损失，为另一个预测创建交叉熵。您可以返回两个损失的加权和作为最终损失。
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/55600106
复制相关文章
相似问题
腾讯云开发者
