matlab计算miou值,【pytorch】图像分割中IOU等评价指标的计算

"""

refer to https://github.com/jfzhang95/pytorch-deeplab-xception/blob/master/utils/metrics.py

"""

import numpy as np

import cv2

__all__ = ['SegmentationMetric']

"""

confusionMetric # 注意：此处横着代表预测值，竖着代表真实值，与之前介绍的相反

P\L P N

P TP FP

N FN TN

"""

class SegmentationMetric(object):

def __init__(self, numClass):

self.numClass = numClass

self.confusionMatrix = np.zeros((self.numClass,) * 2) # 混淆矩阵(空)

def pixelAccuracy(self):

# return all class overall pixel accuracy 正确的像素占总像素的比例

# PA = acc = (TP + TN) / (TP + TN + FP + TN)

acc = np.diag(self.confusionMatrix).sum() / self.confusionMatrix.sum()

return acc

def classPixelAccuracy(self):

# return each cat