Взвешенная бинарная кроссентропия в U-Net (Keras) - PullRequest
0 голосов
/ 22 февраля 2019

В настоящее время я работаю над модифицированной версией в сети U-Net (https://lmb.informatik.uni -freiburg.de / people / ronneber / u-net / ) и пытался реализовать функцию взвешенной двоичной потери кроссентропиив Керасе.

    def weighted_pixelwise_crossentropy(self, wmap):

        def loss(y_true, y_pred):

             return losses.binary_crossentropy(y_true, y_pred) * wmap

        return loss

Хотя эта реализация работает, я не смог увидеть какого-либо влияния на общую точность обучения, проверки и прогнозирования, и поэтому мне интересно, верна ли эта реализация.

Может ли кто-нибудь сказать мне, правильно ли реализована вышеуказанная функция взвешенных потерь или я допустил какую-то ошибку, которую до сих пор не смог распознать?

...