def Frequency_Weighted_Intersection_over_Union(self): """ FWIoU，频权交并比:为MIoU的一种提升，这种方法根据每个类出现的频率为其设置权重。 FWIOU = [(TP+FN)/(TP+FP+TN+FN)] *[TP / (TP + FP + FN)] """ freq = np.sum(self.confusion_matrix, axis=1) / np.sum(self.confusion_matrix) iu = np.diag(self.confusion_matrix) / ( np.sum(self.confusion_matrix, axis=1) + np.sum(self.confusion_matrix, axis=0) - np.diag(self.confusion_matrix)) FWIoU = (freq[freq > 0] * iu[freq > 0]).sum() return FWIoU def addBatch(self, imgPredict, imgLabel): assert imgPredict.shape == imgLabel.shape self.confusionMatrix += self.genConfusionMatrix(imgPredict, imgLabel) # 得到混淆矩阵 return self.confusionMatrix def reset(self): self.confusionMatrix = np.zeros((self.numClass, self.numClass))

这是一个用于计算分割模型性能的评估指标函数。其中，函数`Frequency_Weighted_Intersection_over_Union`是计算频权交并比的函数，该指标是平均交并比的加权平均，其中每个类别的权重由其在数据集中的出现频率决定。函数`addBatch`用于将模型的预测结果和真实标签添加到混淆矩阵中，以便计算性能指标。函数`reset`用于将混淆矩阵重置为全零矩阵，以便在下一次评估之前清空之前的结果。