import os import subprocess import sys from threading import Thread import numpy as np from rscder.gui.actions import ActionManager from rscder.plugins.basic import BasicPlugin from rscder.gui.layercombox import RasterLayerCombox,ResultLayercombox, ResultPointLayerCombox from PyQt5.QtWidgets import QAction, QFileDialog, QDialog, QLabel, QHBoxLayout, QVBoxLayout, QPushButton,QDialogButtonBox,QWidget from PyQt5.QtGui import QIcon from PyQt5.QtCore import Qt from osgeo import gdal from rscder.utils.icons import IconInstance from rscder.utils.project import Project, ResultPointLayer def kappa(confusion_matrix): pe_rows = np.sum(confusion_matrix, axis=0) pe_cols = np.sum(confusion_matrix, axis=1) sum_total = sum(pe_cols) pe = np.dot(pe_rows, pe_cols) / float(sum_total ** 2) po = np.trace(confusion_matrix) / float(sum_total) return (po - pe) / (1 - pe) class EvalutationDialog(QDialog): def __init__(self, parent=None): super().__init__(parent) self.setWindowTitle('精度评估') self.setWindowIcon(IconInstance().LOGO) self.setMinimumWidth(500) self.setMinimumHeight(150) self.layer_select = ResultPointLayerCombox(self) self.gt_file = None gt_file_select_label = QLabel('真值文件:') self.gt_file_select = QPushButton('选择...', self) self.gt_file_select.clicked.connect(self.on_gt_file_select) hbox1 = QHBoxLayout() hbox1.addWidget(gt_file_select_label) hbox1.addWidget(self.gt_file_select) hbox2 = QHBoxLayout() hbox2.addWidget(QLabel('二值化结果图层：')) hbox2.addWidget(self.layer_select) self.ok_button = QPushButton('确定', self) self.ok_button.setIcon(IconInstance().OK) self.ok_button.clicked.connect(self.on_ok) self.ok_button.setDefault(True) self.cancel_button = QPushButton('取消', self) self.cancel_button.setIcon(IconInstance().CANCEL) self.cancel_button.clicked.connect(self.on_cancel) self.cancel_button.setDefault(False) self.buttonbox=QDialogButtonBox(self) self.buttonbox.addButton(self.ok_button,QDialogButtonBox.NoRole) self.buttonbox.addButton(self.cancel_button,QDialogButtonBox.NoRole) self.buttonbox.setCenterButtons(True) self.button_layout = QHBoxLayout() self.button_layout.addWidget(self.ok_button,0,alignment=Qt.AlignHCenter) self.button_layout.addWidget(self.cancel_button,0,alignment=Qt.AlignHCenter) self.main_layout = QVBoxLayout() self.main_layout.addLayout(hbox1) self.main_layout.addLayout(hbox2) self.main_layout.addLayout(self.button_layout) self.setLayout(self.main_layout) def on_gt_file_select(self): file_name, _ = QFileDialog.getOpenFileName(self, '选择真值文件', '', '*.tif') if file_name: self.gt_file = file_name self.gt_file_select.setText(file_name) def on_ok(self): self.accept() def on_cancel(self): self.reject() class EvaluationPlugin(BasicPlugin): @staticmethod def info(): return { 'name': '精度评估', 'author': 'RSC', 'version': '1.0.0', 'description': '精度评估', 'category': 'Evaluation' } def set_action(self): self.action = QAction(IconInstance().EVALUATION, '精度评估', self.mainwindow) self.action.triggered.connect(self.show_dialog) ActionManager().postop_menu.addAction(self.action) def run_alg(self, layer:ResultPointLayer, gt): if layer is None or gt is None: return self.send_message.emit('正在进行精度评估...') pred_ds = gdal.Open(layer.result_path['path']) pred_band = pred_ds.GetRasterBand(1) gt_ds = gdal.Open(gt) gt_band = gt_ds.GetRasterBand(1) if pred_band is None or gt_band is None: return # import pdb # pdb.set_trace() if pred_ds.RasterXSize != gt_ds.RasterXSize or pred_ds.RasterYSize != gt_ds.RasterYSize: self.send_message.emit('真值与预测结果大小不匹配') return xsize = pred_ds.RasterXSize ysize = pred_ds.RasterYSize max_pixels = 1e6 y_block_size = int(max_pixels / xsize) + 1 block_count = int(ysize / y_block_size) + 1 cfm = np.zeros((2,2)) for i in range(block_count): block_size = (xsize, y_block_size) block_offset = (0, i * y_block_size) if i == block_count - 1: block_size = (xsize, ysize - i * y_block_size) pred_block = pred_band.ReadAsArray(*block_offset, *block_size) gt_block = gt_band.ReadAsArray(*block_offset, *block_size) pred_block = pred_block.astype(np.uint8) gt_block = gt_block.astype(np.uint8) valid_mask = ((pred_block == 1) | (pred_block == 0)) & ((gt_block >= 1) | (gt_block == 0)) pred_block = pred_block[valid_mask] gt_block = gt_block[valid_mask] for k in range(2): for l in range(2): # cfm[k,l] += np.sum((pred_block == k) & (gt_block >= l)) cfm[k,l] += np.sum((pred_block == k) & (gt_block == l)) result_path = os.path.join(Project().other_path, f'{layer.name}_{os.path.basename(gt)}_evaluation.txt') with open(result_path, 'w', encoding='utf-8') as f: f.write(f'预测结果：{layer.path}\n') f.write(f'真值： {gt}\n') f.write('混淆矩阵结果：\n') f.write(f''' 真实值 | 变化 | 未变化 预测值 变化 {cfm[1,1]}\t, {cfm[1,0]} 未变化 {cfm[0,1]}\t, {cfm[0,0]} 归一化混淆矩阵： 真实值 | 变化 | 未变化 预测值 变化 {cfm[1,1]/np.sum(cfm)}\t, {cfm[1,0]/np.sum(cfm)} 未变化 {cfm[0,1]/np.sum(cfm)}\t, {cfm[0,0]/np.sum(cfm)} OA: {(cfm[1,1] + cfm[0,0])/np.sum(cfm)} F1: {cfm[1,1] / (np.sum(cfm[1,:]) + np.sum(cfm[:,1]) - cfm[1,1])} Kappa: {kappa(cfm)} ''') f.flush() # os.system('pause') txt=f''' 预测结果：{layer.path}\n 真值： {gt}\n 混淆矩阵结果：\n 真实值 | 变化 | 未变化 预测值 变化 {cfm[1,1]}\t, {cfm[1,0]} 未变化 {cfm[0,1]}\t, {cfm[0,0]} 归一化混淆矩阵： 真实值 | 变化 | 未变化 预测值 变化 {cfm[1,1]/np.sum(cfm)}\t, {cfm[1,0]/np.sum(cfm)} 未变化 {cfm[0,1]/np.sum(cfm)}\t, {cfm[0,0]/np.sum(cfm)} OA: {(cfm[1,1] + cfm[0,0])/np.sum(cfm)} F1: {cfm[1,1] / (np.sum(cfm[1,:]) + np.sum(cfm[:,1]) - cfm[1,1])} Kappa: {kappa(cfm)} ''' # self.msgbox = Win(txt) # self.msgbox.show() # os.system(f'c:/windows/notepad.exe "{result_path}"') self.send_message.emit('精度评估完成,结果保存在'+result_path) def show_dialog(self): dialog = EvalutationDialog(self.mainwindow) dialog.exec_() if dialog.result() == QDialog.Accepted: layer = dialog.layer_select.current_layer # if not isinstance(layer, SingleBandRasterLayer): # self.send_message.emit('请选择一个单波段栅格图层') # return t = Thread(target=self.run_alg, args=(layer, dialog.gt_file)) t.start() from .window import Ui_Form # class Win(QWidget,Ui_Form): # def __init__(self,txt): # super().__init__() # self.setupUi(self) # self.label.setText(txt)