227 lines
8.1 KiB
Python
227 lines
8.1 KiB
Python
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)
|
|
|
|
|
|
|