语义分割准确率计算

目录

pytorch版

pytorch准确率，miou：

sklearn版

---

pytorch版

"""
reference from: https://github.com/LeeJunHyun/Image_Segmentation/blob/master/evaluation.py
"""
 
import torch
 
# SR : Segmentation Result
# GT : Ground Truth
 
def get_accuracy(SR,GT,threshold=0.5):
    SR = SR > threshold
    GT = GT == torch.max(GT)
    corr = torch.sum(SR==GT)
    tensor_size = SR.size(0)*SR.size(1)*SR.size(2)*SR.size(3)
    acc = float(corr)/float(tensor_size)
 
    return acc
 
def get_sensitivity(SR,GT,threshold=0.5):
    # Sensitivity == Recall
    SR = SR > threshold
    GT = GT == torch.max(GT)
 
    # TP : True Positive
    # FN : False Negative
    TP = ((SR==1)+(GT==1))==2
    FN = ((SR==0)+(GT==1))==2
 
    SE = float(torch.sum(TP))/(float(torch.sum(TP+FN)) + 1e-6)