mIOU公式在此不赘述。
1、fast_hist(label.flatten(), pred.flatten(), num_classes)
2、生成关于像素的混淆矩阵
k = (a >= 0) & (a < n)
这句用来过滤噪声像素
np.bincount(n*a[k].astype(int)+b[k],minlength=n**2).reshape(n,n)
解释如下:
(1) n*a[k].astype(int)+b[k]
中n=2,a= label.flatten(),b= pred.flatten()
python 图像语义分割mIOU代码详解
最新推荐文章于 2023-05-11 10:20:49 发布