SSIM / MS-SSIM для TensorFlow

Существует ли SSIM или даже MS-SSIM реализация для TensorFlow ?

SSIM ( показатель структурного сходства индекса ) представляет собой показатель для измерения качества изображения или сходства изображений. Он вдохновлен человеческим восприятием и, согласно нескольким статьям, гораздо лучше выполняет функцию потерь по сравнению с l1 / l2. Например, см. Функции потерь для нейронных сетей для обработки изображений .

До сих пор я не мог найти реализацию в TensorFlow. И после попытки сделать это самостоятельно, портировав его из C ++ или кода Python (такого как Github: VQMT / SSIM ), я застрял на таких методах, как применение размытия по Гауссу к изображению в TensorFlow.

Кто-то уже пытался реализовать это сам?