ndimage.grey_erosion NotImplementedError：无法将符号张量ArgMax：0转换为numpy数组

如何解决ndimage.grey_erosion NotImplementedError：无法将符号张量ArgMax：0转换为numpy数组

from scipy import ndimage

def teknofest_IoU(y_true,y_pred):
    y_true = tf.argmax(y_true,axis=-1)
    y_pred = tf.argmax(y_pred,axis=-1)

    mask_erosion = ndimage.grey_erosion(y_true,footprint=np.ones((3,3)))
    mask_dilation = ndimage.grey_dilation(y_true,3)))
    
    tp,tn,fn,fp = 0,0

    y_pred = K.flatten(y_pred)
    mask_erosion = K.flatten(mask_erosion)
    mask_dilation = K.flatten(mask_dilation)

    for i in range(y_pred.shape[0]):
        # case 0: y_pred[i] = 0
        # case 1: y_pred[i] = 1
        # case 2: y_pred[i] = 2
        if y_pred[i] != 0 and mask_dilation[i] == y_pred[i]:
            tp += 1
        elif mask_erosion[i] != 0 and y_pred[i] == mask_erosion[i]:
            fn += 1
        elif y_pred[i] == 0:
            tn += 1
        else:
            fp += 1

    IoU = tp / (fp + tp + fn)

    return IoU

这是我的代码，我认为它会出错，因为 grey_dilation 试图将张量转换为 numpy 数组，但我无法替代 ndimage.grey_erosion 和 ndimage.grey_dilation。我该如何解决这个问题？