def dice_coef(y_true, y_pred):
y_pred_decision = tf.floor(y_pred / K.max(y_pred, axis=4, keepdims=True))
y_true_f = K.flatten(y_true)
y_pred_f = K.flatten(y_pred_decision)
intersection = K.sum(y_true_f * y_pred_f)
return (2. * intersection + K.epsilon()) / (K.sum(y_true_f) + K.sum(y_pred_f) + K.epsilon())
评论列表
文章目录