• 欢迎访问开心洋葱网站,在线教程,推荐使用最新版火狐浏览器和Chrome浏览器访问本网站,欢迎加入开心洋葱 QQ群
  • 为方便开心洋葱网用户,开心洋葱官网已经开启复制功能!
  • 欢迎访问开心洋葱网站,手机也能访问哦~欢迎加入开心洋葱多维思维学习平台 QQ群
  • 如果您觉得本站非常有看点,那么赶紧使用Ctrl+D 收藏开心洋葱吧~~~~~~~~~~~~~!
  • 由于近期流量激增,小站的ECS没能经的起亲们的访问,本站依然没有盈利,如果各位看如果觉着文字不错,还请看官给小站打个赏~~~~~~~~~~~~~!

numpy图像语义分隔评估实现

人工智能 _寒潭雁影 2569次浏览 0个评论

在做图像语言分隔的时候,我们往往需要计算出像素精确度,交并比等值来评估我们的算法或者神经网络的表现。本篇文章着重介绍如何用numpy实现像素精确度、平均像素精确度、平均交并比和加权的交并比的计算。   在正式开始之前需要先掌握几个numpy的函数和编程技巧。  

bincount函数

功能:实现个数的统计。 示例:假设有numpy数组:  

x = np.array([0, 1, 1, 3, 2, 1, 7])

  我们想要知道里面0~7这8个数出现的次数是多少,就可以用bincount函数。  

y=np.bincount(x)

  打印出y得到[1 3 1 1 0 0 0 1],意思是我们有0~7这8个数,按照数组下标排列起来,然后对应的数值则为0~7出现的次数。 需要注意的是这里的数值长度是函数自己给的,也就是说它只会统计到我们放进函数的这个数组中最大的那个值。如果我们想要统计0~9这10个数出现的次数,我们可以通过指定minlength参数实现:  

y = np.bincount(x,minlength=10)

  输出则为:[1 3 1 1 0 0 0 1 0 0]

diag函数

功能:取出对角线上的值。 示例:假设有numpy数组:  

x=np.array([(1,0,0,0,0,4),(0,2,0,2,0,0),(0,0,3,0,0,4),(0,0,4,4,0,0),(0,0,5,0,5,4),(0,0,0,2,0,6)])

  [[1 0 0 0 0 4] [0 2 0 2 0 0] [0 0 3 0 0 4] [0 0 4 4 0 0] [0 0 5 0 5 4] [0 0 0 2 0 6]] 我们想要取出对角线上的数,则可以用diag函数。  

y=np.diag(x)

  输出为:[1 2 3 4 5 6] 如果我们想要把对角线上的数加起来,则写成:  

y=np.diag(x).sum()

    输出为:21

zip函数

功能:依据多维数组的第一维进行遍历。 示例:假设有numpy数组:  

x=np.array([[(0,0,0,0,0,4), (0,0,0,2,0,0)],[(0,0,0,0,0,4), (0,0,0,2,0,0)]])
y=np.array([[(0,0,0,0,0,4), (0,0,0,2,0,0)],[(0,0,0,0,0,1), (0,0,0,2,0,0)]])

  它们的形状是:(2, 2, 6) 假设前面的2指代的是不同的图片,而后面的2和6指代的是图片的宽和高 ,而我们希望将图片一张张取出来进行计算,则可以用zip函数来做。  

for onex, oney in zip(x, y):
    print(onex.shape)
    print(oney.shape)

 

flatten函数

  功能:将多维数组直接转化成1维 示例:假设有numpy数组:  

x=np.array([[(0,0,0,0,0,4), (0,0,0,2,0,0)],[(0,0,0,0,0,4), (0,0,0,2,0,0)]])

  我们希望把数组转为1维的,则可以用flatten函数。  

y=x.flatten()

  输出为:[0 0 0 0 0 4 0 0 0 2 0 0 0 0 0 0 0 4 0 0 0 2 0 0]

数组感兴趣区域选取

功能:选取感兴趣值进行计算,其他非感兴趣区域置为0 示例:假设有numpy数组:  

x = np.array([0, 1, 1, 3, 2, 1, 7])

  我们只想计算0~2这个3个数,而无视其他的数,则可以用如下代码实现:  

mask = (x >= 0) & (x <= 2)
x=x[mask]

  输出为:[0 1 1 2 1] 可以看到,非感兴趣区域被删除掉了。 上面就是所有需要先掌握的点,下面直接放出图像语义分隔评估的实现代码  

import numpy as np

# 精确度计算
def _fast_hist(label_true, label_pred, n_class):
    #标注计算类别范围以内的数,不在范围内的不考虑
    mask = (label_true >= 0) & (label_true < n_class)
    # 经过这句话的处理,行数将代表标签的值,而列数将代表预测的值,而具体的值代表个数。
    # 也就是说,对角线上的数分别指代第n个类别中,预测值和标签值相同的个数
    hist = np.bincount(n_class * label_true[mask].astype(int) +label_pred[mask], 
            minlength=n_class ** 2).reshape(n_class, n_class)
    return hist

def label_accuracy_score(label_trues, label_preds, n_class):
    """Returns accuracy score evaluation result.
      - overall accuracy
      - mean accuracy
      - mean IU
      - fwavacc
    """
    hist = np.zeros((n_class, n_class))
    #将标签和预测图像按行放入函数中
    for lt, lp in zip(label_trues, label_preds):
        hist += _fast_hist(lt.flatten(), lp.flatten(), n_class)
    acc = np.diag(hist).sum() / hist.sum()#总体精确度计算
    acc_cls = np.diag(hist) / hist.sum(axis=1)#求得每一类像素精确度的平均值
    acc_cls = np.nanmean(acc_cls)#求得总体平均精确度
    iu = np.diag(hist) / (hist.sum(axis=1) + hist.sum(axis=0) - np.diag(hist))#按类别求得交并比
    mean_iu = np.nanmean(iu)#求得总体平均交并比
    freq = hist.sum(axis=1) / hist.sum()#求得标签中各个类别所占的比例
    fwavacc = (freq[freq > 0] * iu[freq > 0]).sum()#按权值计算交并比
    return acc, acc_cls, mean_iu, fwavacc

label_true=np.array([[(0,0,0,0,0,4), (0,0,0,2,0,0)],[(0,0,0,0,0,4), (0,0,0,2,0,0)]])
label_pred=np.array([[(0,0,0,0,0,4), (0,0,0,2,0,0)],[(0,0,0,0,0,1), (0,0,0,2,0,0)]])

train_acc = 0
train_acc_cls = 0
train_mean_iu = 0 
train_fwavacc = 0

#用zip可以遍历多维数组的第一个维度,在本应用中每次返回一张标签和一张预测图像
for lbt, lbp in zip(label_true, label_pred):
    acc, acc_cls, mean_iu, fwavacc = label_accuracy_score(lbt, lbp, 5)#5为要计算的类别总数,也就是0~4
    train_acc += acc
    train_acc_cls += acc_cls
    train_mean_iu += mean_iu
    train_fwavacc += fwavacc

 


开心洋葱 , 版权所有丨如未注明 , 均为原创丨未经授权请勿修改 , 转载请注明numpy图像语义分隔评估实现
喜欢 (0)

您必须 登录 才能发表评论!

加载中……