问题描述
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。我该如何解决这个问题?
解决方法
暂无找到可以解决该程序问题的有效方法,小编努力寻找整理中!
如果你已经找到好的解决方法,欢迎将解决方案带上本链接一起发送给小编。
小编邮箱:dio#foxmail.com (将#修改为@)