Нужно ли преобразовывать маски предсказания в двоичную форму?

Question

Я тренирую U- net для двоичной сегментации. Вопрос в том, хочу ли я использовать потерю игральных костей. Должен ли я использовать бинаризованные маски предсказания перед их использованием в этой функции?

def dice_coef(y_true, y_pred):
    y_true_f = K.flatten(y_true)
    y_pred_f = K.flatten(y_pred)
    intersection = K.sum(y_true_f * y_pred_f)
    return (2. * intersection + 1.) / (K.sum(y_true_f) + K.sum(y_pred_f) + 1.)


def dice_coef_loss(y_true, y_pred):
    return 1-dice_coef(y_true, y_pred)

Функция активации в последнем слое - сигмовидная функция.