使用flow_from_directory时骰子系数> 1

时间:2018-08-29 03:17:18

标签: python-2.7 keras-2 unity3d-unet

我正在使用UNet进行医学图像分割。由于内存问题,我想加载图像以使用flow_from_directory训练模型。问题是骰子系数的计算> 1。

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 + smooth) / (K.sum(y_true_f) + K.sum(y_pred_f) +smooth)

来自目录的流,如keras文档示例中所示 https://keras.io/preprocessing/image/#imagedatagenerator-class

0 个答案:

没有答案