python中如何计算分割边界与地面真值边界的均方距离

2024-06-13 22:35:47 发布

您现在位置:Python中文网/ 问答频道 /正文

我有两个图像,一个是地面真实图像,另一个是分割图像。我想计算一下这两者之间的MSD。我写了一个代码。只是想知道下面的代码是否正确。你知道吗

grd=np.array(measure.find_contours(gt, 0.99))
grd_x=np.array(np.transpose(grd[:,:,0]))
grd_y=np.array(np.transpose(grd[:,:,1]))

seg=np.array(measure.find_contours(a,0.99))
seg_x=np.array(np.transpose(seg[:,:,0]))
seg_y=np.array(np.transpose(seg[:,:,1]))

distance=[]
for i in range(0,len(seg_x)):
     temp_vector_x=seg_x[i]*np.ones(len(grd_x))
     temp_vector_y=seg_y[i]*np.ones(len(grd_y))
     tt=np.array(np.sqrt((np.square(temp_vector_x-grd_x))+(np.square(temp_vector_y-grd_y))))
     ttmin=tt.min()
     distance.append(ttmin)

results=(np.sum(distance))/len(seg_x)

Tags: 代码图像lennponesfindarraytemp