Мне нужна SSIM как функция потерь в моей сети, но моя сеть имеет 2 выхода. Мне нужно использовать SSIM для первого вывода и cross-entropy
для следующего. Функция потерь представляет собой их комбинацию. Однако мне нужно иметь более высокий SSIM и более низкий cross-entropy
, поэтому я думаю, что их комбинация неверна. Другая проблема в том, что я не смог найти реализацию SSIM в keras.
В Tensorflow есть tf.image.ssim
, но он принимает изображение, и я не думаю, что смогу использовать его в функции потерь, верно? Не могли бы вы сказать мне, что мне делать? Я новичок в керасе и глубоком обучении, и я не знаю, как сделать SSIM в качестве пользовательской функции потерь в керасе.