如何在python中计算分割边界与地面实况边界的均方距离

时间:2017-10-22 06:36:38

标签: python

我有两张图像说 gt 地面实况图像和 a 分割图像。我想计算这两者之间的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)

0 个答案:

没有答案