Я делаю семантическую сегментацию изображения.В большинстве случаев количество пикселей переднего плана намного меньше, чем количество пикселей фона изображения.
Я обнаружил, что кто-то опубликовал это в моем предыдущем посте.Но так как я новичок в Keras, я не знаю, как реализовать эту функцию.Мой вопрос, где в моем коде мне нужно реализовать / вызвать это?это в model.fit_generator?
def pixel_wise_loss(y_true, y_pred):
pos_weight = tf.constant([[1.0, 2.0]])
loss = tf.nn.weighted_cross_entropy_with_logits(
y_true,
y_pred,
pos_weight,
name=None
)
return K.mean(loss,axis=-1)
Извините, если я задал этот глупый вопрос, но я новичок в Keras и все еще учусь.Заранее спасибо!Было бы здорово, если бы кто-то мог опубликовать пример кода о том, как его реализовать.