添加项目文件。

DEM_ASSESS.sln

@@ -1,22 +1,22 @@
 
 Microsoft Visual Studio Solution File, Format Version 12.00
 # Visual Studio Version 16
-VisualStudioVersion = 16.0.32630.194
+VisualStudioVersion = 16.0.32126.315
 MinimumVisualStudioVersion = 10.0.40219.1
-Project("{8BC9CEB8-8B4A-11D0-8D11-00A0C91BC942}") = "LandslideTrain", "LandslideTrain\LandslideTrain.vcxproj", "{1D23BAC7-1D1A-48BE-9796-93F34E435A5A}"
+Project("{8BC9CEB8-8B4A-11D0-8D11-00A0C91BC942}") = "DEM_ASSESS", "DEM_ASSESS\DEM_ASSESS.vcxproj", "{E901DBCF-219B-4AA9-A34F-BBDDFC06FE11}"
 EndProject
 Global
 	GlobalSection(SolutionConfigurationPlatforms) = preSolution
 		Release|x64 = Release|x64
 	EndGlobalSection
 	GlobalSection(ProjectConfigurationPlatforms) = postSolution
-		{1D23BAC7-1D1A-48BE-9796-93F34E435A5A}.Release|x64.ActiveCfg = Release|x64
+		{E901DBCF-219B-4AA9-A34F-BBDDFC06FE11}.Release|x64.ActiveCfg = Release|x64
-		{1D23BAC7-1D1A-48BE-9796-93F34E435A5A}.Release|x64.Build.0 = Release|x64
+		{E901DBCF-219B-4AA9-A34F-BBDDFC06FE11}.Release|x64.Build.0 = Release|x64
 	EndGlobalSection
 	GlobalSection(SolutionProperties) = preSolution
 		HideSolutionNode = FALSE
 	EndGlobalSection
 	GlobalSection(ExtensibilityGlobals) = postSolution
-		SolutionGuid = {9DAFCFF4-8408-40F6-A160-95047A99BB28}
+		SolutionGuid = {1945C514-1E36-458A-841E-E9B4B69F86D4}
 	EndGlobalSection
 EndGlobal

DEM_ASSESS/DEM_ASSESS.cpp

@@ -0,0 +1,386 @@
+#include "DEM_ASSESS.h"
+
+#include <QSettings>
+
+void WorkThreadObject::runAssessWork(QString model, QString PRE, QString label, QString index, QString result)
+{
+	qDebug() << "start work:" << "\npredict: " << PRE << "\nLabel:" << label;
+	QFileInfo file(index);
+	file.absolutePath();
+	assess_txt_path = file.absolutePath() + "/assessment.txt";
+	//QDir pluginsDir = QDir(qApp->applicationDirPath());
+	QDir pluginsDir = QDir("D:\\code\\RsSurvey\\RsSurvey_Build\\x64\\Release\\");
+	qDebug() << "----" << pluginsDir.currentPath();
+	if (!pluginsDir.cd("models\\envs"))
+	{
+		qDebug() << "no folder models\\envs";
+		return;
+	}
+	QString exeDirName = pluginsDir.absoluteFilePath("miou_vegetation.exe");
+	QString model_dir = " --model_dir " + model;
+	QString img_data_dir = " --img_data_dir " + PRE + "/";
+	QString label_dir = " --label_dir " + label + "/";
+	QString val_list = " --val_list " + index;
+	QString out_result = " --result_file " + result + "/";
+	QString ss = exeDirName + model_dir + img_data_dir + label_dir + val_list + out_result;
+	qDebug() << ss;
+	QProcess* pProces = new QProcess(this);
+	connect(pProces, SIGNAL(readyReadStandardOutput()), this, SLOT(writeAssessTxt()));
+	pProces->start(ss);
+
+	emit process(2);
+}
+
+void WorkThreadObject::writeAssessTxt()
+{
+	mProces = (QProcess*)sender();
+	QString output = QString::fromLocal8Bit(mProces->readAllStandardOutput());
+	output.chop(2);
+	if (output.toFloat() != 0)
+	{
+		qDebug() << "process:" << output.toFloat();
+		emit process(output.toFloat());
+	}
+	else
+		qDebug() << "unsolved exe out:" << output;
+}
+
+void WorkThreadObject::on_cancel()
+{
+	if (mProces == nullptr)
+	{
+		qDebug() << "--mProces  null";
+	}
+	else
+	{
+		QString KillStr = "taskkill /f /im miou_vegetation.exe";
+		mProces->startDetached(KillStr);
+		qDebug() << "--kill Proces";
+	}
+}
+///////////////////////////////////////////////////////
+
+ModelAssessment::ModelAssessment()
+{
+	//ui.setupUi(this);
+}
+
+ModelAssessment::~ModelAssessment()
+{
+
+}
+
+QString ModelAssessment::PannelName()
+{
+	return QString::fromLocal8Bit("");
+}
+
+QString ModelAssessment::CategoryName()
+{
+	return QString::fromLocal8Bit("DEM模块");
+}
+
+QString ModelAssessment::EnglishName()
+{
+	return QString::fromLocal8Bit("DEM_Module");
+}
+
+QString ModelAssessment::ChineseName()
+{
+	return QString::fromLocal8Bit("结果评估");
+}
+
+QString ModelAssessment::Information()
+{
+	return "结果评估";
+}
+
+QString ModelAssessment::IconPath()
+{
+	return ":/DEM_ASSESS/resources/assessment.svg";
+}
+
+QWidget* ModelAssessment::CenterWidget()
+{
+	QString gdal_path = qApp->applicationDirPath().toLocal8Bit() + "/share/gdal";
+	QString pro_lib_path = qApp->applicationDirPath().toLocal8Bit() + "/share/proj";
+	qputenv("GDAL_DATA", gdal_path.toLocal8Bit());
+	qputenv("PROJ_LIB", pro_lib_path.toLocal8Bit());
+
+	bool showWin = false;
+	if (myWidget == nullptr)
+	{
+		myWidget = new QDialog();
+		showWin = true;
+		qDebug() << "new QDialog()";
+	}
+	else
+	{
+		qDebug() << "already have myWidget";
+		//激活窗口并提升至顶层
+		myWidget->activateWindow();
+		myWidget->raise();
+		return myWidget;
+	}
+	ui.setupUi(myWidget);
+
+	myWidget->setWindowTitle(QString::fromLocal8Bit("结果评估"));
+	myWidget->setWindowFlags(Qt::CustomizeWindowHint | Qt::WindowCloseButtonHint);
+	myWidget->setWindowIcon(QIcon(":/DEM_ASSESS/resources/assessment.svg"));
+	myWidget->setAttribute(Qt::WA_QuitOnClose, false);
+
+	myWidget->setAttribute(Qt::WA_DeleteOnClose);
+	connect(myWidget, &QDialog::destroyed, this, [=] {
+		qDebug() << "----DEM assess window close----";
+
+		QDir pluginsDir = QDir(qApp->applicationDirPath());
+		if (pluginsDir.cd("srsplugins\\DemModel"))
+		{
+			QString strConfigPath = pluginsDir.absoluteFilePath("dem_config.ini");
+			QFile f(strConfigPath);
+			if (f.exists())
+			{
+				WriteConfigPaths(strConfigPath);
+				f.close();
+			}
+		}
+
+		if (mWorkThread != nullptr)
+		{
+			//mWorkThread->quit();
+			//mWorkThread->wait();//调用wait后先调用finished信号对应的槽函数，执行完成后再往下走
+			//mWorkThread = nullptr;//mWorkThread, &QThread::finished, mWorkThread, &QObject::deleteLater，不需要delete
+			//mWorker = nullptr;//mWorkThread, &QThread::finished, mWorker, &QObject::deleteLater
+			mWorkThread->requestInterruption();//请求线程中断
+			mWorkThread->quit();
+			mWorkThread->wait();//调用wait后先调用finished信号对应的槽函数，执行完成后再往下走
+			mWorker->on_cancel();
+			mWorkThread = nullptr;//mWorkThread, &QThread::finished, mWorkThread, &QObject::deleteLater，不需要delete
+			mWorker = nullptr;//mWorkThread, &QThread::finished, mWorker, &QObject::deleteLater
+		}
+		//qDebug() << "--destroyed";
+		myWidget->close();
+		myWidget = nullptr;
+		});
+	connect(ui.pbtInPredict, &QPushButton::pressed, this, &ModelAssessment::choosePredictPath);
+	connect(ui.pbtInLabel, &QPushButton::pressed, this, &ModelAssessment::chooseLabelPath);
+	connect(ui.pbtIndex, &QPushButton::pressed, this, &ModelAssessment::chooseAssessFile);
+	connect(ui.pbtModel, &QPushButton::pressed, this, &ModelAssessment::chooseModelFile);
+	connect(ui.pbtOutResult, &QPushButton::pressed, this, &ModelAssessment::chooseResultPath);
+
+	connect(ui.pushButton_ok, &QPushButton::pressed, this, &ModelAssessment::readAndStart);
+	connect(ui.pushButton_cancel, &QPushButton::pressed, this, &ModelAssessment::pbCancel);
+
+	ui.progressBar->setTextVisible(true);
+	ui.progressBar->setRange(0, 100);
+
+	QFile qssFile(":/DEM_ASSESS/DEM_ASSESS.qss");
+	qssFile.open(QFile::ReadOnly); //以只读方式打开
+	if (qssFile.isOpen())
+	{
+		QString qss = QLatin1String(qssFile.readAll());
+		myWidget->setStyleSheet(qss);
+		qssFile.close();
+	}
+	else
+		qDebug() << "无法打开文件";
+
+	QDir pluginsDir = QDir(qApp->applicationDirPath());
+	if (pluginsDir.cd("srsplugins\\DemModel"))
+	{
+		QString strConfigPath = pluginsDir.absoluteFilePath("dem_config.ini");
+		QFile f(strConfigPath);
+		if (f.exists())
+		{
+			ReadConfigHistoryPaths(strConfigPath);
+			f.close();
+		}
+	}
+
+	if (showWin)
+		myWidget->show();
+
+	return myWidget;
+}
+
+void ModelAssessment::startWorkThread()
+{
+	if (mWorker != nullptr)
+	{
+		return;
+	}
+	mWorkThread = new QThread();
+	mWorker = new WorkThreadObject();
+	mWorker->moveToThread(mWorkThread);
+	connect(mWorkThread, &QThread::finished, mWorkThread, &QObject::deleteLater);
+	connect(mWorkThread, &QThread::finished, mWorker, &QObject::deleteLater);
+	connect(mWorker, &WorkThreadObject::process, myWidget, [=](int value) {
+		ui.progressBar->setValue(value);
+		if (value == 100)
+		{
+			QString txtDir = ui.lineOutResult->text() + "/assessment.txt";
+			QFile file(txtDir);
+			QString outLines = "";
+			if (file.open(QIODevice::ReadOnly))
+			{
+				QTextStream stream(&file);
+				// recall 召回率
+				// precision 精确率
+				// mIoU 平均交并比
+				// mAP 平均精度均值
+				// f1_score
+				while (!stream.atEnd())
+				{
+					QString line = stream.readLine();
+					//strs.push_back(line);
+					if (line.contains("===>mIoU"))
+					{
+						//===>mIoU:68.45; mAP: 74.92
+						line.remove("===>");
+						line.replace(QStringLiteral("mAP"), QStringLiteral("平均精度均值mAP"));
+						line.replace(QStringLiteral("mIoU"), QStringLiteral("平均交并比mIoU"));
+						outLines = outLines + line + "\n";
+					}
+					else if (line.contains("===>recall"))
+					{
+						//===>recall:66.62; precision: 51.2
+						line.remove("===>");
+						line.replace(QStringLiteral("recall"), QStringLiteral("召回率recall"));
+						line.replace(QStringLiteral("precision"), QStringLiteral("精确率precision"));
+						outLines = outLines + line + "\n";
+					}
+					else if (line.contains("===>f1_score"))
+					{
+						//===>f1_score:57.9
+						line.remove("===>");
+						line.replace(QStringLiteral("f1_score"), QStringLiteral("F1分数f1_score"));
+						outLines = outLines + line;
+					}
+				}
+				file.close();
+			}
+
+			QMessageBox mess(QMessageBox::Information,
+				QString::fromLocal8Bit("评估结束"),
+				QString::fromLocal8Bit("评价结果:\n") + txtDir + "\n"
+				+ outLines);
+			mess.setWindowFlags(Qt::Drawer);
+			mess.setTextInteractionFlags(Qt::TextSelectableByMouse);
+			mess.setStandardButtons(QMessageBox::Yes);
+			mess.button(QMessageBox::StandardButton::Yes)->setText(QString::fromLocal8Bit("确认"));
+			//mess.setTextInteractionFlags(Qt::TextSelectableByMouse);
+			int result = mess.exec();
+			pbCancel();
+		}
+		});
+	connect(this, &ModelAssessment::startPreAssessment, mWorker, &WorkThreadObject::runAssessWork);
+
+	mWorkThread->start();
+}
+
+void ModelAssessment::chooseModelFile()
+{
+	QString fileModel = QFileDialog::getOpenFileName(ui.pbtModel, QString::fromLocal8Bit("选择输入模型文件"), "", "*.pth");
+	if (fileModel != "")
+		ui.lineModel->setText(fileModel);
+}
+void ModelAssessment::choosePredictPath()
+{
+	QString dirDom = QFileDialog::getExistingDirectory(ui.pbtInPredict, QString::fromLocal8Bit("选择输入评价文件路径"), "");
+	if (dirDom != "")
+		ui.lineInPredict->setText(dirDom);
+}
+void ModelAssessment::chooseLabelPath()
+{
+	QString dirDom = QFileDialog::getExistingDirectory(ui.pbtInLabel, QString::fromLocal8Bit("选择输入Label文件路径"), "");
+	if (dirDom != "")
+		ui.lineInLabel->setText(dirDom);
+}
+void ModelAssessment::chooseAssessFile()
+{
+	QString dirDom = QFileDialog::getOpenFileName(ui.pbtIndex, QString::fromLocal8Bit("选择输入评估文件索引"), "", "*.txt");
+	if (dirDom != "")
+		ui.lineIndex->setText(dirDom);
+}
+void ModelAssessment::chooseResultPath()
+{
+	QString dirResult = QFileDialog::getExistingDirectory(ui.pbtOutResult, QString::fromLocal8Bit("选择评估结果输出路径"), "");
+	if (dirResult != "")
+		ui.lineOutResult->setText(dirResult);
+}
+
+void ModelAssessment::ReadConfigHistoryPaths(QString strPath)
+{
+	QSettings configIni(strPath, QSettings::IniFormat);
+
+	//打开标题为：[DemTrain] 的组，读取DemTrain输出结果TrainResult字段
+	configIni.beginGroup("DemTrain");
+
+	QString strDemTrainResult = configIni.value("TrainResult").toString();
+	QDir srcDir = QDir(strDemTrainResult);
+	//更新评价文件列表
+	if (srcDir.absoluteFilePath("val_list.txt") != "")
+		ui.lineIndex->setText(srcDir.absoluteFilePath("val_list.txt"));
+	//更新输入Images
+	srcDir = QDir(strDemTrainResult);
+	if (srcDir.cd("Images"))
+		ui.lineInPredict->setText(srcDir.absolutePath());
+	//更新输入Labels
+	srcDir = QDir(strDemTrainResult);
+	if (srcDir.cd("Labels"))
+		ui.lineInLabel->setText(srcDir.absolutePath());
+
+	configIni.endGroup();//关闭组
+
+	//打开标题为：[DemAssess] 的组，读取AssessResult字段
+	configIni.beginGroup("DemAssess");
+	//更新评价结果输出
+	ui.lineOutResult->setText(configIni.value("AssessResult").toString());
+
+	configIni.endGroup();//关闭组
+}
+
+void ModelAssessment::WriteConfigPaths(QString strPath)
+{
+	QSettings configIni(strPath, QSettings::IniFormat);
+	configIni.setIniCodec("utf-8");
+	//打开标题为：[DemAssess] 的组
+	configIni.beginGroup("DemAssess");
+
+	//更新输入模型路径
+	QString temp = ui.lineOutResult->text();
+	if (temp != "")
+		configIni.setValue("AssessResult", temp);
+
+	configIni.endGroup();//关闭组
+}
+
+void ModelAssessment::readAndStart()
+{
+	QString dirModel = ui.lineModel->text();
+	QString dirIndex = ui.lineIndex->text();
+	QString dirLabel = ui.lineInLabel->text();
+	QString dirPredict = ui.lineInPredict->text();
+	QString dirResult = ui.lineOutResult->text();
+
+	ui.progressBar->setValue(0);
+
+	if (dirLabel == "" || dirPredict == "" || dirIndex == "" || dirModel == "" || dirResult == "")
+	{
+		QMessageBox mess(QMessageBox::NoIcon, QString::fromLocal8Bit("错误"), QString::fromLocal8Bit("请检查输入路径"));
+		mess.setWindowFlags(Qt::Drawer);
+		int result = mess.exec();
+		return;
+	}
+	if (mWorkThread == nullptr)
+	{
+		qDebug() << "--startThread";
+		startWorkThread();
+	}
+	emit startPreAssessment(dirModel, dirPredict, dirLabel, dirIndex, dirResult);
+}
+
+void ModelAssessment::pbCancel()
+{
+	delete myWidget;//调起&QDialog::destroyed
+}

DEM_ASSESS/DEM_ASSESS.h

@@ -0,0 +1,79 @@
+#pragma once
+
+#include <QtWidgets>
+#include <QtWidgets/QDialog>
+#include <QThread>
+#include <QFile>
+#include <QMessageBox>
+
+#include "ui_DEM_ASSESS.h"
+#include "SrsMainPluginInterFace.h"
+
+class WorkThreadObject :public QObject
+{
+	Q_OBJECT
+
+public:
+	void on_cancel();
+
+signals:
+	void process(int value);
+
+public slots:
+	void runAssessWork(QString model, QString PRE, QString label, QString index, QString result);
+	void writeAssessTxt();
+
+private:
+	QProcess* mProces = nullptr;
+	QString assess_txt_path;
+};
+
+class ModelAssessment : public SrsMainInterface
+{
+	Q_OBJECT
+		Q_INTERFACES(SrsMainInterface)
+		Q_PLUGIN_METADATA(IID MainInterface_iid)
+
+public:
+	ModelAssessment();
+	~ModelAssessment();
+
+	virtual QString PannelName() override;
+	virtual QString CategoryName() override;
+
+	virtual QString EnglishName() override;
+	virtual QString ChineseName() override;
+	virtual QString Information() override;
+	virtual QString IconPath() override;
+	virtual QWidget* CenterWidget() override;
+
+	//读json文件，获取历史存储的路径
+	void ReadConfigHistoryPaths(QString strPath);
+	//保存本次打开的路径到json文件
+	void WriteConfigPaths(QString strPath);
+
+	void startWorkThread();
+
+signals:
+	void startPreAssessment(QString model, QString PRE, QString label, QString index, QString result);
+
+public slots:
+	//pushbutton_OK
+	void readAndStart();
+	void pbCancel();
+
+	void choosePredictPath();
+	void chooseLabelPath();
+	void chooseAssessFile();
+	void chooseModelFile();
+	void chooseResultPath();
+
+private:
+	Ui::DEM_ASSESSClass ui;
+
+	QDialog* myWidget = nullptr;
+
+	WorkThreadObject* mWorker = nullptr;
+	QThread* mWorkThread = nullptr;
+
+};

DEM_ASSESS/DEM_ASSESS.qrc

@@ -0,0 +1,6 @@
+<RCC>
+    <qresource prefix="/DEM_ASSESS">
+        <file>DEM_ASSESS.qss</file>
+        <file>resources/assessment.svg</file>
+    </qresource>
+</RCC>

DEM_ASSESS/DEM_ASSESS.qss

@@ -22,7 +22,7 @@ QLabel{
 }
 
 /*----QPushButtonÑùʽ±í*/
-#pushButton_ok, #pushButton_cancel, #pbtInModel, #pbtInDataset, #pbtInDsm, #pbtInLabel, #pbtOutResult{
+#pushButton_ok, #pushButton_cancel, #pbtInPredict, #pbtInLabel, #pbtModel, #pbtIndex,#pbtOutResult{
 	font-family:'Microsoft YaHei';
 	font-size:12px;
 	background-color: #ffffff;
@@ -34,35 +34,35 @@ QLabel{
 #pushButton_ok, #pushButton_cancel{
 	min-width:36px;
 }
-#pushButton_ok:hover, #pushButton_cancel:hover, #pbtInModel:hover, #pbtInDataset:hover, #pbtInLabel:hover,#pbtOutResult:hover{
+#pushButton_ok:hover, #pushButton_cancel:hover, #pbtInPredict:hover, #pbtInLabel:hover, #pbtModel:hover, #pbtIndex:hover,#pbtOutResult:hover{
 	background-color: #ecf5ff;
 	color: #409eff;
 }
-#pushButton_ok:pressed, #pushButton_cancel:pressed, #pbtInModel:pressed, #pbtInDataset:pressed, #pbtInLabel:pressed,#pbtOutResult:pressed{
+#pushButton_ok:pressed, #pushButton_cancel:pressed, #pbtInPredict:pressed, #pbtInLabel:pressed, #pbtModel:pressed, #pbtIndex:pressed,#pbtOutResult:pressed{
 	border: 1px solid #3a8ee6;
 	color: #409eff;
 }
-#pushButton_ok:checked, #pushButton_cancel:checked, #pbtInModel:checked, #pbtInDataset:checked, #pbtInLabel:checked,#pbtOutResult:checked{
+#pushButton_ok:checked, #pushButton_cancel:checked, #pbtInPredict:checked, #pbtInLabel:checked, #pbtModel:checked, #pbtIndex:checked,#pbtOutResult:checked{
 	border: 1px solid #3a8ee6;
 	color: #409eff;
 }
-#pushButton_ok:focus, #pushButton_cancel:focus, #pbtInModel:focus, #pbtInDataset:focus, #pbtInLabel:focus,#pbtOutResult:focus{
+#pushButton_ok:focus, #pushButton_cancel:focus, #pbtInPredict:focus, #pbtInLabel:focus, #pbtModel:focus, #pbtIndex:focus,#pbtOutResult:focus{
 	border: 1px solid #3a8ee6;
 	color: #409eff;
 	outline: none;
 }
 
 /*----QLineEditÑùʽ*/
-#lineInModel, #lineInDataset, #lineOutResult, #lineInLabel, #lineEpoch1, #lineEpoch2{
+#lineInPredict, #lineInLabel, #lineModel, #lineIndex, #lineOutResult{
 	border:0px;
 	border-bottom: 1px solid #B3B3B3;
 	font-family:'Microsoft YaHei';
 	font-size:12px;
 	background-color:transparent;
 }
-#lineInModel:hover, #lineInDataset:hover, #lineOutResult:hover, #lineInLabel:hover, #lineEpoch1:hover, #lineEpoch2:hover{
+#lineInPredict:hover, #lineInLabel:hover, #lineModel:hover, #lineIndex:hover, #lineOutResult:hover{
 	border-bottom: 2px solid #66A3FF;
 }
-#lineInModel:focus, #lineInDataset:focus, #lineOutResult:focus, #lineInLabel:focus, #lineEpoch1:focus, #lineEpoch2:focus{
+#lineInPredict:focus, #lineInLabel:focus, #lineModel:focus, #lineIndex:focus, #lineOutResult:focus{
 	border-bottom: 2px solid #7666FF;
 }

DEM_ASSESS/DEM_ASSESS.ui

@@ -1,7 +1,7 @@
 <?xml version="1.0" encoding="UTF-8"?>
 <ui version="4.0">
- <class>LandslideTrainClass</class>
+ <class>DEM_ASSESSClass</class>
- <widget class="QDialog" name="LandslideTrainClass">
+ <widget class="QDialog" name="DEM_ASSESSClass">
   <property name="geometry">
    <rect>
     <x>0</x>
@@ -23,34 +23,175 @@
    </size>
   </property>
   <property name="windowTitle">
-   <string>LandslideTrain</string>
+   <string>DEM_ASSESS</string>
+  </property>
+  <property name="windowIcon">
+   <iconset>
+    <normaloff>resources/assessment.svg</normaloff>resources/assessment.svg</iconset>
   </property>
   <layout class="QGridLayout" name="gridLayout_2">
    <item row="0" column="0">
     <layout class="QGridLayout" name="gridLayout">
-     <item row="3" column="1">
+     <item row="2" column="0">
-      <widget class="QLineEdit" name="lineOutResult"/>
+      <widget class="QLabel" name="label_mask_2">
-     </item>
-     <item row="0" column="1">
-      <widget class="QLineEdit" name="lineInModel"/>
-     </item>
-     <item row="1" column="0">
-      <widget class="QLabel" name="label_2">
        <property name="sizePolicy">
-        <sizepolicy hsizetype="Preferred" vsizetype="Expanding">
+        <sizepolicy hsizetype="Minimum" vsizetype="Expanding">
          <horstretch>0</horstretch>
          <verstretch>0</verstretch>
         </sizepolicy>
        </property>
+       <property name="minimumSize">
+        <size>
+         <width>0</width>
+         <height>0</height>
+        </size>
+       </property>
+       <property name="maximumSize">
+        <size>
+         <width>16777215</width>
+         <height>16777215</height>
+        </size>
+       </property>
        <property name="text">
-        <string>输入训练数据路径</string>
+        <string>输入验证文件路径</string>
+       </property>
+       <property name="alignment">
+        <set>Qt::AlignRight|Qt::AlignTrailing|Qt::AlignVCenter</set>
        </property>
       </widget>
      </item>
+     <item row="3" column="0">
+      <widget class="QLabel" name="label_mask">
+       <property name="sizePolicy">
+        <sizepolicy hsizetype="Minimum" vsizetype="Expanding">
+         <horstretch>0</horstretch>
+         <verstretch>0</verstretch>
+        </sizepolicy>
+       </property>
+       <property name="minimumSize">
+        <size>
+         <width>0</width>
+         <height>0</height>
+        </size>
+       </property>
+       <property name="maximumSize">
+        <size>
+         <width>16777215</width>
+         <height>16777215</height>
+        </size>
+       </property>
+       <property name="text">
+        <string>输入标签文件路径</string>
+       </property>
+       <property name="alignment">
+        <set>Qt::AlignRight|Qt::AlignTrailing|Qt::AlignVCenter</set>
+       </property>
+      </widget>
+     </item>
+     <item row="0" column="1">
+      <widget class="QLineEdit" name="lineModel"/>
+     </item>
      <item row="1" column="1">
-      <widget class="QLineEdit" name="lineInDataset"/>
+      <widget class="QLineEdit" name="lineIndex"/>
+     </item>
+     <item row="2" column="1">
+      <widget class="QLineEdit" name="lineInPredict">
+       <property name="sizePolicy">
+        <sizepolicy hsizetype="Expanding" vsizetype="Fixed">
+         <horstretch>0</horstretch>
+         <verstretch>0</verstretch>
+        </sizepolicy>
+       </property>
+      </widget>
+     </item>
+     <item row="0" column="0">
+      <widget class="QLabel" name="label_2">
+       <property name="sizePolicy">
+        <sizepolicy hsizetype="Minimum" vsizetype="Expanding">
+         <horstretch>0</horstretch>
+         <verstretch>0</verstretch>
+        </sizepolicy>
+       </property>
+       <property name="maximumSize">
+        <size>
+         <width>16777215</width>
+         <height>16777215</height>
+        </size>
+       </property>
+       <property name="text">
+        <string>输入模型文件路径</string>
+       </property>
+       <property name="alignment">
+        <set>Qt::AlignRight|Qt::AlignTrailing|Qt::AlignVCenter</set>
+       </property>
+      </widget>
+     </item>
+     <item row="1" column="2">
+      <widget class="QPushButton" name="pbtIndex">
+       <property name="text">
+        <string>选择文件</string>
+       </property>
+      </widget>
      </item>
      <item row="2" column="2">
+      <widget class="QPushButton" name="pbtInPredict">
+       <property name="minimumSize">
+        <size>
+         <width>75</width>
+         <height>0</height>
+        </size>
+       </property>
+       <property name="maximumSize">
+        <size>
+         <width>75</width>
+         <height>16777215</height>
+        </size>
+       </property>
+       <property name="text">
+        <string>选择文件夹</string>
+       </property>
+      </widget>
+     </item>
+     <item row="3" column="1">
+      <widget class="QLineEdit" name="lineInLabel">
+       <property name="sizePolicy">
+        <sizepolicy hsizetype="Expanding" vsizetype="Fixed">
+         <horstretch>0</horstretch>
+         <verstretch>0</verstretch>
+        </sizepolicy>
+       </property>
+      </widget>
+     </item>
+     <item row="1" column="0">
+      <widget class="QLabel" name="label">
+       <property name="sizePolicy">
+        <sizepolicy hsizetype="Minimum" vsizetype="Expanding">
+         <horstretch>0</horstretch>
+         <verstretch>0</verstretch>
+        </sizepolicy>
+       </property>
+       <property name="maximumSize">
+        <size>
+         <width>16777215</width>
+         <height>16777215</height>
+        </size>
+       </property>
+       <property name="text">
+        <string>输入验证文件索引</string>
+       </property>
+       <property name="alignment">
+        <set>Qt::AlignRight|Qt::AlignTrailing|Qt::AlignVCenter</set>
+       </property>
+      </widget>
+     </item>
+     <item row="0" column="2">
+      <widget class="QPushButton" name="pbtModel">
+       <property name="text">
+        <string>选择文件</string>
+       </property>
+      </widget>
+     </item>
+     <item row="3" column="2">
       <widget class="QPushButton" name="pbtInLabel">
        <property name="minimumSize">
         <size>
@@ -65,43 +206,11 @@
         </size>
        </property>
        <property name="text">
-        <string>打开文件夹</string>
+        <string>选择文件夹</string>
        </property>
       </widget>
      </item>
-     <item row="0" column="2">
+     <item row="4" column="0">
-      <widget class="QPushButton" name="pbtInModel">
-       <property name="minimumSize">
-        <size>
-         <width>75</width>
-         <height>0</height>
-        </size>
-       </property>
-       <property name="maximumSize">
-        <size>
-         <width>75</width>
-         <height>16777215</height>
-        </size>
-       </property>
-       <property name="text">
-        <string>打开文件</string>
-       </property>
-      </widget>
-     </item>
-     <item row="0" column="0">
-      <widget class="QLabel" name="label">
-       <property name="sizePolicy">
-        <sizepolicy hsizetype="Preferred" vsizetype="Expanding">
-         <horstretch>0</horstretch>
-         <verstretch>0</verstretch>
-        </sizepolicy>
-       </property>
-       <property name="text">
-        <string>输入初始训练模型路径</string>
-       </property>
-      </widget>
-     </item>
-     <item row="2" column="0">
       <widget class="QLabel" name="label_3">
        <property name="sizePolicy">
         <sizepolicy hsizetype="Preferred" vsizetype="Expanding">
@@ -110,46 +219,14 @@
         </sizepolicy>
        </property>
        <property name="text">
-        <string>输入标签数据路径</string>
+        <string>评估结果输出路径</string>
        </property>
       </widget>
      </item>
-     <item row="3" column="0">
+     <item row="4" column="1">
-      <widget class="QLabel" name="label_4">
+      <widget class="QLineEdit" name="lineOutResult"/>
-       <property name="sizePolicy">
-        <sizepolicy hsizetype="Preferred" vsizetype="Expanding">
-         <horstretch>0</horstretch>
-         <verstretch>0</verstretch>
-        </sizepolicy>
-       </property>
-       <property name="text">
-        <string>输出模型文件路径</string>
-       </property>
-      </widget>
      </item>
-     <item row="1" column="2">
+     <item row="4" column="2">
-      <widget class="QPushButton" name="pbtInDataset">
-       <property name="minimumSize">
-        <size>
-         <width>75</width>
-         <height>0</height>
-        </size>
-       </property>
-       <property name="maximumSize">
-        <size>
-         <width>75</width>
-         <height>16777215</height>
-        </size>
-       </property>
-       <property name="text">
-        <string>打开文件夹</string>
-       </property>
-      </widget>
-     </item>
-     <item row="2" column="1">
-      <widget class="QLineEdit" name="lineInLabel"/>
-     </item>
-     <item row="3" column="2">
       <widget class="QPushButton" name="pbtOutResult">
        <property name="minimumSize">
         <size>
@@ -164,51 +241,7 @@
         </size>
        </property>
        <property name="text">
-        <string>打开文件夹</string>
+        <string>选择文件夹</string>
-       </property>
-      </widget>
-     </item>
-     <item row="4" column="1">
-      <layout class="QHBoxLayout" name="horizontalLayout_2">
-       <item>
-        <widget class="QLineEdit" name="lineEpoch1">
-         <property name="maximumSize">
-          <size>
-           <width>80</width>
-           <height>16777215</height>
-          </size>
-         </property>
-        </widget>
-       </item>
-       <item>
-        <widget class="QLabel" name="label_6">
-         <property name="text">
-          <string>解冻训练次数</string>
-         </property>
-         <property name="alignment">
-          <set>Qt::AlignRight|Qt::AlignTrailing|Qt::AlignVCenter</set>
-         </property>
-        </widget>
-       </item>
-       <item>
-        <widget class="QLineEdit" name="lineEpoch2">
-         <property name="maximumSize">
-          <size>
-           <width>80</width>
-           <height>16777215</height>
-          </size>
-         </property>
-        </widget>
-       </item>
-      </layout>
-     </item>
-     <item row="4" column="0">
-      <widget class="QLabel" name="label_5">
-       <property name="text">
-        <string>冻结训练次数</string>
-       </property>
-       <property name="alignment">
-        <set>Qt::AlignRight|Qt::AlignTrailing|Qt::AlignVCenter</set>
        </property>
       </widget>
      </item>
@@ -216,29 +249,6 @@
    </item>
    <item row="1" column="0">
     <layout class="QHBoxLayout" name="horizontalLayout">
-     <property name="leftMargin">
-      <number>40</number>
-     </property>
-     <item>
-      <widget class="QCheckBox" name="checkBoxGenData">
-       <property name="text">
-        <string>生成训练集、验证集</string>
-       </property>
-       <property name="checked">
-        <bool>true</bool>
-       </property>
-      </widget>
-     </item>
-     <item>
-      <widget class="QCheckBox" name="checkBoxTrain">
-       <property name="text">
-        <string>模型训练</string>
-       </property>
-       <property name="checked">
-        <bool>true</bool>
-       </property>
-      </widget>
-     </item>
      <item>
       <spacer name="horizontalSpacer">
        <property name="orientation">
@@ -254,8 +264,14 @@
      </item>
      <item>
       <widget class="QPushButton" name="pushButton_ok">
+       <property name="sizePolicy">
+        <sizepolicy hsizetype="Minimum" vsizetype="Fixed">
+         <horstretch>0</horstretch>
+         <verstretch>0</verstretch>
+        </sizepolicy>
+       </property>
        <property name="text">
-        <string>确认</string>
+        <string>确定</string>
        </property>
       </widget>
      </item>
@@ -269,7 +285,7 @@
        </property>
        <property name="sizeHint" stdset="0">
         <size>
-         <width>10</width>
+         <width>13</width>
          <height>20</height>
         </size>
        </property>
@@ -285,7 +301,7 @@
     </layout>
    </item>
    <item row="2" column="0">
-    <widget class="CProgressBar" name="progressBar">
+    <widget class="QProgressBar" name="progressBar">
      <property name="value">
       <number>0</number>
      </property>
@@ -294,31 +310,8 @@
   </layout>
  </widget>
  <layoutdefault spacing="6" margin="11"/>
- <customwidgets>
-  <customwidget>
-   <class>CProgressBar</class>
-   <extends>QProgressBar</extends>
-   <header>CProgressBar.h</header>
-  </customwidget>
- </customwidgets>
- <tabstops>
-  <tabstop>lineInModel</tabstop>
-  <tabstop>pbtInModel</tabstop>
-  <tabstop>lineInDataset</tabstop>
-  <tabstop>pbtInDataset</tabstop>
-  <tabstop>lineInLabel</tabstop>
-  <tabstop>pbtInLabel</tabstop>
-  <tabstop>lineOutResult</tabstop>
-  <tabstop>pbtOutResult</tabstop>
-  <tabstop>lineEpoch1</tabstop>
-  <tabstop>lineEpoch2</tabstop>
-  <tabstop>checkBoxGenData</tabstop>
-  <tabstop>checkBoxTrain</tabstop>
-  <tabstop>pushButton_ok</tabstop>
-  <tabstop>pushButton_cancel</tabstop>
- </tabstops>
  <resources>
-  <include location="LandslideTrain.qrc"/>
+  <include location="DEM_ASSESS.qrc"/>
  </resources>
  <connections/>
 </ui>

DEM_ASSESS/DEM_ASSESS.vcxproj

@@ -7,14 +7,14 @@
     </ProjectConfiguration>
   </ItemGroup>
   <PropertyGroup Label="Globals">
-    <ProjectGuid>{1D23BAC7-1D1A-48BE-9796-93F34E435A5A}</ProjectGuid>
+    <ProjectGuid>{E901DBCF-219B-4AA9-A34F-BBDDFC06FE11}</ProjectGuid>
     <Keyword>QtVS_v304</Keyword>
-    <WindowsTargetPlatformVersion Condition="'$(Configuration)|$(Platform)' == 'Release|x64'">10.0</WindowsTargetPlatformVersion>
+    <WindowsTargetPlatformVersion Condition="'$(Configuration)|$(Platform)' == 'Release|x64'">10.0.19041.0</WindowsTargetPlatformVersion>
     <QtMsBuild Condition="'$(QtMsBuild)'=='' OR !Exists('$(QtMsBuild)\qt.targets')">$(MSBuildProjectDirectory)\QtMsBuild</QtMsBuild>
   </PropertyGroup>
   <Import Project="$(VCTargetsPath)\Microsoft.Cpp.Default.props" />
   <PropertyGroup Condition="'$(Configuration)|$(Platform)' == 'Release|x64'" Label="Configuration">
-    <ConfigurationType>DynamicLibrary</ConfigurationType>
+    <ConfigurationType>Application</ConfigurationType>
     <PlatformToolset>v142</PlatformToolset>
   </PropertyGroup>
   <Import Project="$(VCTargetsPath)\Microsoft.Cpp.props" />
@@ -37,9 +37,7 @@
   </ImportGroup>
   <PropertyGroup Label="UserMacros" />
   <PropertyGroup Condition="'$(Configuration)|$(Platform)' == 'Release|x64'">
-    <IncludePath>D:\qgis\osgeo4w\include;D:\qgis\osgeo4w\apps\Qt5\include;$(IncludePath)</IncludePath>
+    <TargetName>dem2-$(ProjectName)</TargetName>
-    <LibraryPath>D:\qgis\osgeo4w\lib;D:\qgis\osgeo4w\apps\Qt5\lib;$(LibraryPath)</LibraryPath>
-    <TargetName>landslide1_train</TargetName>
   </PropertyGroup>
   <ItemDefinitionGroup Condition="'$(Configuration)|$(Platform)' == 'Release|x64'" Label="Configuration">
     <ClCompile>
@@ -48,29 +46,30 @@
       <DebugInformationFormat>EditAndContinue</DebugInformationFormat>
       <Optimization>Disabled</Optimization>
       <RuntimeLibrary>MultiThreadedDLL</RuntimeLibrary>
+      <UsePrecompiledHeader>Use</UsePrecompiledHeader>
+      <PrecompiledHeaderFile>stdafx.h</PrecompiledHeaderFile>
     </ClCompile>
     <Link>
       <SubSystem>Console</SubSystem>
       <GenerateDebugInformation>true</GenerateDebugInformation>
     </Link>
+    <QtMoc>
+      <PrependInclude>%(PrependInclude)</PrependInclude>
+    </QtMoc>
   </ItemDefinitionGroup>
   <ItemGroup>
-    <QtRcc Include="LandslideTrain.qrc" />
+    <QtRcc Include="DEM_ASSESS.qrc" />
-    <QtUic Include="LandslideTrain.ui" />
+    <QtUic Include="DEM_ASSESS.ui" />
-    <QtMoc Include="LandslideTrain.h" />
+    <QtMoc Include="DEM_ASSESS.h" />
-    <ClCompile Include="CProgressBar.cpp" />
+    <ClCompile Include="DEM_ASSESS.cpp" />
-    <ClCompile Include="LandslideTrain.cpp" />
     <ClCompile Include="main.cpp" />
   </ItemGroup>
   <ItemGroup>
     <QtMoc Include="SrsMainPluginInterFace.h" />
+    <ClInclude Include="x64\Release\uic\ui_DEM_ASSESS.h" />
   </ItemGroup>
   <ItemGroup>
-    <QtMoc Include="CProgressBar.h" />
+    <None Include="DEM_ASSESS.qss" />
-    <ClInclude Include="x64\Release\uic\ui_LandslideTrain.h" />
-  </ItemGroup>
-  <ItemGroup>
-    <None Include="LandslideTrain.qss" />
   </ItemGroup>
   <Import Project="$(VCTargetsPath)\Microsoft.Cpp.targets" />
   <ImportGroup Condition="Exists('$(QtMsBuild)\qt.targets')">

DEM_ASSESS/DEM_ASSESS.vcxproj.filters

@@ -3,7 +3,7 @@
   <ItemGroup>
     <Filter Include="Source Files">
       <UniqueIdentifier>{4FC737F1-C7A5-4376-A066-2A32D752A2FF}</UniqueIdentifier>
-      <Extensions>qml;cpp;c;cc;cxx;def;odl;idl;hpj;bat;asm;asmx</Extensions>
+      <Extensions>cpp;c;cc;cxx;def;odl;idl;hpj;bat;asm;asmx</Extensions>
     </Filter>
     <Filter Include="Header Files">
       <UniqueIdentifier>{93995380-89BD-4b04-88EB-625FBE52EBFB}</UniqueIdentifier>
@@ -23,16 +23,16 @@
     </Filter>
   </ItemGroup>
   <ItemGroup>
-    <QtRcc Include="LandslideTrain.qrc">
+    <QtRcc Include="DEM_ASSESS.qrc">
       <Filter>Resource Files</Filter>
     </QtRcc>
-    <QtUic Include="LandslideTrain.ui">
+    <QtUic Include="DEM_ASSESS.ui">
       <Filter>Form Files</Filter>
     </QtUic>
-    <QtMoc Include="LandslideTrain.h">
+    <QtMoc Include="DEM_ASSESS.h">
       <Filter>Header Files</Filter>
     </QtMoc>
-    <ClCompile Include="LandslideTrain.cpp">
+    <ClCompile Include="DEM_ASSESS.cpp">
       <Filter>Source Files</Filter>
     </ClCompile>
   </ItemGroup>
@@ -40,25 +40,19 @@
     <ClCompile Include="main.cpp">
       <Filter>Source Files</Filter>
     </ClCompile>
-    <ClCompile Include="CProgressBar.cpp">
+  </ItemGroup>
-      <Filter>Source Files</Filter>
+  <ItemGroup>
-    </ClCompile>
+    <ClInclude Include="x64\Release\uic\ui_DEM_ASSESS.h">
+      <Filter>Form Files</Filter>
+    </ClInclude>
   </ItemGroup>
   <ItemGroup>
     <QtMoc Include="SrsMainPluginInterFace.h">
       <Filter>Header Files</Filter>
     </QtMoc>
-    <QtMoc Include="CProgressBar.h">
-      <Filter>Header Files</Filter>
-    </QtMoc>
   </ItemGroup>
   <ItemGroup>
-    <ClInclude Include="x64\Release\uic\ui_LandslideTrain.h">
+    <None Include="DEM_ASSESS.qss">
-      <Filter>Form Files</Filter>
-    </ClInclude>
-  </ItemGroup>
-  <ItemGroup>
-    <None Include="LandslideTrain.qss">
       <Filter>Resource Files</Filter>
     </None>
   </ItemGroup>

DEM_ASSESS/main.cpp

@@ -0,0 +1,16 @@
+#include "DEM_ASSESS.h"
+#include <QtWidgets/QApplication>
+
+int main(int argc, char *argv[])
+{
+	QApplication a(argc, argv);
+
+	QString gdal_path = qApp->applicationDirPath().toLocal8Bit() + "/share/gdal";
+	QString pro_lib_path = qApp->applicationDirPath().toLocal8Bit() + "/share/proj";
+	qputenv("GDAL_DATA", gdal_path.toLocal8Bit());
+	qputenv("PROJ_LIB", pro_lib_path.toLocal8Bit());
+
+	ModelAssessment w;
+	w.CenterWidget()->show();
+	return a.exec();
+}

DEM_ASSESS/resources/assessment.svg

@@ -0,0 +1,2 @@
+<?xml version="1.0" standalone="no"?><!DOCTYPE svg PUBLIC "-//W3C//DTD SVG 1.1//EN" "http://www.w3.org/Graphics/SVG/1.1/DTD/svg11.dtd"><svg t="1652932649746" class="icon" viewBox="0 0 1024 1024" version="1.1" xmlns="http://www.w3.org/2000/svg" p-id="2186" width="32" height="32" xmlns:xlink="http://www.w3.org/1999/xlink"><defs><style type="text/css">@font-face { font-family: feedback-iconfont; src: url("//at.alicdn.com/t/font_1031158_u69w8yhxdu.woff2?t=1630033759944") format("woff2"), url("//at.alicdn.com/t/font_1031158_u69w8yhxdu.woff?t=1630033759944") format("woff"), url("//at.alicdn.com/t/font_1031158_u69w8yhxdu.ttf?t=1630033759944") format("truetype"); }
+</style></defs><path d="M802.3 649c-13.3 0-24 10.7-24 24v163.9c0 17.7-14.4 32.1-32.1 32.1H222.6c-17.7 0-32.1-14.4-32.1-32.1v-656c0-17.7 14.4-32.1 32.1-32.1h523.7c17.7 0 32 14.4 32 32V235c0 13.3 10.7 24 24 24s24-10.7 24-24v-54.2c0-44.1-35.9-80-80-80H222.6c-44.2 0-80.1 35.9-80.1 80.1v656c0 44.2 35.9 80.1 80.1 80.1h523.6c44.2 0 80.1-35.9 80.1-80.1V673c0-13.3-10.7-24-24-24z" fill="#2c2c2c" p-id="2187"></path><path d="M856.6 332.2L827 315.1c-15.2-9.3-31.5-11.1-47-5.2-12.7 4.8-24.3 14.8-33.8 29l-16.4 25.2c-0.8 0.8-1.5 1.7-2.1 2.6-0.7 1-1.2 2.1-1.7 3.2L563 620.5c-8.8 13.2-13.2 29.2-12.6 45.1l-2.2 90.4c-0.1 5.7 5.4 9.7 10.8 7.9l100.8-35.2c15.6-4.6 28.6-14.8 36.8-28.9L866 411.2l0.1-0.1c5.3-9.1 11.7-23.2 12.1-38 0.5-17.5-7.2-32-21.6-40.9z m-50.3 17.1l29.6 17.1c1.4 0.9 2.4 1.7 2.3 5.4-0.1 5-2.6 12-6.7 19.1l-8.5 14.5-50.6-33.3 7.3-11.2c13.1-19.8 22.6-14 26.6-11.6zM662.1 679.5l-0.1 0.1c-3 5.2-8 9-13.8 10.7l-0.5 0.1-51.7 18c-3.2 1.1-6.6-1.3-6.5-4.7l1-38.7v-0.8c-0.4-7.6 1.7-15.4 5.9-21.7l154.2-236.9 52.2 34.3-140.7 239.6zM702.6 256.9c-7.8-7.8-20.5-7.9-28.3-0.1L467.8 462.4l-93.9-98.6c-7-7.4-18.5-8.3-26.6-2.1l-113.1 86.6c-8.8 6.7-10.4 19.3-3.7 28 6.7 8.8 19.3 10.4 28 3.7l98.8-75.7 95.6 100.4c3.7 3.9 8.8 6.1 14.2 6.2h0.3c5.3 0 10.4-2.1 14.1-5.8l221.1-220c7.8-7.7 7.8-20.4 0-28.2zM266.3 647h186.2c11 0 20-9 20-20s-9-20-20-20H266.3c-11 0-20 9-20 20s9 20 20 20zM266.3 775h232.2c11 0 20-9 20-20s-9-20-20-20H266.3c-11 0-20 9-20 20s9 20 20 20z" fill="#2c2c2c" p-id="2188"></path></svg>

LandslideTrain/CProgressBar.cpp

@@ -1,51 +0,0 @@
-#include "CProgressBar.h"
-
-CProgressBar::CProgressBar(QWidget* parent) : QProgressBar(parent)
-{
-	//this->setStyleSheet(
-	//	"QProgressBar {color:black;font-size:12px;font-weight:normal;font-family:'Microsoft YaHei'; text-align:center; }"
-	//	"QProgressBar::chunk {background-color: rgb(0, 160, 244);}");
-}
-
-void CProgressBar::ShowTips(const QString& strTips)
-{
-	if (this->isHidden())
-	{
-		this->show();
-	}
-
-	this->resetFormat();
-	this->setRange(0, 100);
-	this->setFormat(strTips);
-	this->setValue(100);
-}
-
-double CProgressBar::GetDoubleFormatValue()
-{
-	double dValue = 0.0;
-
-	QString strFormat = this->format();
-	if (!strFormat.isEmpty())
-	{
-		int nIndex = strFormat.indexOf(":");
-		if (nIndex != -1)
-		{
-			QString strFormatData = strFormat.mid(nIndex + 1, strFormat.length() - nIndex - 2);
-			dValue = strFormatData.toDouble();
-		}
-	}
-
-	return dValue;
-}
-
-void CProgressBar::SetDoubleFormatValue(const QString& strFormat, double dValue)
-{
-	if (this->isHidden())
-	{
-		this->show();
-	}
-
-	this->setFormat(strFormat + ":" + QString::number(dValue, 'f', 2) + "%");
-	this->setValue(dValue);
-}
-

LandslideTrain/CProgressBar.h

@@ -1,18 +0,0 @@
-#pragma once
-#include<QProgressBar>
-
-class CProgressBar : public QProgressBar
-{
-	Q_OBJECT
-
-public:
-	CProgressBar(QWidget* parent = 0);
-	~CProgressBar() {}
-
-public:
-	void ShowTips(const QString& strTips);
-
-	void SetDoubleFormatValue(const QString& strFormat, double dValue);
-
-	double GetDoubleFormatValue();
-};

LandslideTrain/LandslideTrain.cpp

@@ -1,455 +0,0 @@
-#include "LandslideTrain.h"
-
-#include <QSettings>
-
-LandslideTrain::LandslideTrain()
-{
-	//ui.setupUi(this);
-}
-
-QString LandslideTrain::PannelName()
-{
-	return QString::fromLocal8Bit("地质模块");
-}
-
-QString LandslideTrain::CategoryName()
-{
-	return QString::fromLocal8Bit("地质模块");
-}
-
-QString LandslideTrain::EnglishName()
-{
-	return QString::fromLocal8Bit("LandslideTrain");
-}
-
-QString LandslideTrain::ChineseName()
-{
-	return QString::fromLocal8Bit("模型构建");
-}
-
-QString LandslideTrain::Information()
-{
-	return QString::fromLocal8Bit("模型构建");
-}
-
-QString LandslideTrain::IconPath()
-{
-	return ":/LandslideTrain/resources/dem_vec.svg";
-}
-
-QWidget* LandslideTrain::CenterWidget()
-{
-	//QString gdal_path = qApp->applicationDirPath().toLocal8Bit() + "/share/gdal";
-	//QString pro_lib_path = qApp->applicationDirPath().toLocal8Bit() + "/share/proj";
-	//qputenv("GDAL_DATA", gdal_path.toLocal8Bit());
-	//qputenv("PROJ_LIB", pro_lib_path.toLocal8Bit());
-	bool showWin = false;
-	if (myWidget == nullptr)
-	{
-		myWidget = new QDialog();
-		showWin = true;
-		qDebug() << "new QDialog()";
-	}
-	else
-	{
-		qDebug() << "already have myWidget";
-		//激活窗口并提升至顶层
-		myWidget->activateWindow();
-		myWidget->raise();
-		return myWidget;
-	}
-	ui.setupUi(myWidget);
-
-	myWidget->setWindowTitle(QString::fromLocal8Bit("模型训练"));
-	myWidget->setWindowFlags(Qt::CustomizeWindowHint | Qt::WindowCloseButtonHint);
-	myWidget->setWindowIcon(QIcon(":/LandslideTrain/resources/dem_vec.svg"));
-	myWidget->setAttribute(Qt::WA_QuitOnClose, false);
-
-	myWidget->setAttribute(Qt::WA_DeleteOnClose);
-	connect(myWidget, &QDialog::destroyed, this, [=] {
-		qDebug() << "----Landslide train window close----";
-
-		QDir pluginsDir = QDir(qApp->applicationDirPath());
-		if (pluginsDir.cd("srsplugins\\SldModel"))
-		{
-			QString strConfigPath = pluginsDir.absoluteFilePath("sld_config.ini");
-			QFile f(strConfigPath);
-			if (f.exists())
-			{
-				WriteConfigPaths(strConfigPath);
-				f.close();
-			}
-		}
-
-		if (mWorkThread != nullptr)
-		{
-			mWorkThread->requestInterruption();
-			mWorkThread->quit();
-			mWorkThread->wait();//调用wait后先调用finished信号对应的槽函数，执行完成后再往下走
-			mWorkObject->on_cancel();
-			mWorkThread = nullptr;//mWorkThread, &QThread::finished, mWorkThread, &QObject::deleteLater，不需要delete
-			mWorkObject = nullptr;//mWorkThread, &QThread::finished, mWorker, &QObject::deleteLater
-		}
-
-		myWidget->close();
-		myWidget = nullptr;
-		});
-	connect(ui.pbtInModel, &QPushButton::clicked, this, &LandslideTrain::chooseInModel);
-	connect(ui.pbtInDataset, &QPushButton::clicked, this, &LandslideTrain::chooseInDataset);
-	connect(ui.pbtInLabel, &QPushButton::clicked, this, &LandslideTrain::chooseInLabel);
-	connect(ui.pbtOutResult, &QPushButton::clicked, this, &LandslideTrain::chooseResultPath);
-
-	connect(ui.pushButton_ok, &QPushButton::clicked, this, &LandslideTrain::readAndStart);
-	connect(ui.pushButton_cancel, &QPushButton::clicked, this, &LandslideTrain::pbCancel);
-
-	ui.lineEpoch1->setValidator(new QIntValidator(0, 999, this));
-	ui.lineEpoch2->setValidator(new QIntValidator(0, 999, this));
-	ui.lineEpoch1->setText("100");
-	ui.lineEpoch2->setText("100");
-
-	ui.pbtInModel->setFocus();
-
-	connect(ui.checkBoxGenData, &QCheckBox::clicked, this, [=](bool checked) {
-		executeGenData = checked;
-		if (!executeGenData && !executeTrain)
-			ui.pushButton_ok->setEnabled(false);
-		else
-			ui.pushButton_ok->setEnabled(true);
-		});
-	connect(ui.checkBoxTrain, &QCheckBox::clicked, this, [=](bool checked) {
-		executeTrain = checked;
-		if (!executeGenData && !executeTrain)
-			ui.pushButton_ok->setEnabled(false);
-		else
-			ui.pushButton_ok->setEnabled(true);
-		});
-
-	ui.progressBar->setTextVisible(true);
-	ui.progressBar->setRange(0, 100);
-
-	QFile qssFile(":/LandslideTrain/LandslideTrain.qss");
-	qssFile.open(QFile::ReadOnly); //以只读方式打开
-	if (qssFile.isOpen())
-	{
-		QString qss = QLatin1String(qssFile.readAll());
-		myWidget->setStyleSheet(qss);
-		qssFile.close();
-	}
-	else
-		qDebug() << "-- no qssFile";
-
-	QDir pluginsDir = QDir(qApp->applicationDirPath());
-	if (pluginsDir.cd("srsplugins\\SldModel"))
-	{
-		QString strConfigPath = pluginsDir.absoluteFilePath("sld_config.ini");
-		QFile f(strConfigPath);
-		if (f.exists())
-		{
-			ReadConfigHistoryPaths(strConfigPath);
-			f.close();
-		}
-	}
-
-	if (showWin)
-		myWidget->show();
-
-	return myWidget;
-}
-
-void LandslideTrain::startWorkThread()
-{
-	if (mWorkObject != nullptr)
-	{
-		return;
-	}
-	mWorkThread = new QThread();
-	mWorkObject = new WorkObject();
-	mWorkObject->moveToThread(mWorkThread);
-	connect(mWorkThread, &QThread::finished, mWorkThread, &QObject::deleteLater);
-	connect(mWorkThread, &QThread::finished, mWorkObject, &QObject::deleteLater);
-	connect(mWorkObject, &WorkObject::progress, myWidget, [=](double val) {ui.progressBar->SetDoubleFormatValue(QString::fromLocal8Bit("进度"), val); });
-	connect(mWorkObject, &WorkObject::trainFinished, this, &LandslideTrain::finished);
-
-	connect(this, &LandslideTrain::start, mWorkObject, &WorkObject::runTrainWork);
-	mWorkThread->start();
-}
-
-void LandslideTrain::ReadConfigHistoryPaths(QString strPath)
-{
-	QSettings configIni(strPath, QSettings::IniFormat);
-
-	//打开标题为：[SldTrain] 的组，并读取出port字段的值
-	configIni.beginGroup("SldTrain");
-
-	ui.lineInModel->setText(configIni.value("RetrainModel").toString());
-	ui.lineInDataset->setText(configIni.value("SrcDom").toString());
-	ui.lineInLabel->setText(configIni.value("SrcLabel").toString());
-	ui.lineOutResult->setText(configIni.value("TrainResult").toString());
-
-	configIni.endGroup();//关闭组
-}
-
-void LandslideTrain::WriteConfigPaths(QString strPath)
-{
-	QSettings configIni(strPath, QSettings::IniFormat);
-	configIni.setIniCodec("utf-8");
-	//打开标题为：[SldTrain] 的组
-	configIni.beginGroup("SldTrain");
-
-	//更新输入模型路径
-	QString temp = ui.lineInModel->text();
-	if (temp != "")
-		configIni.setValue("RetrainModel", temp);
-	//更新输入DOM路径
-	temp = ui.lineInDataset->text();
-	if (temp != "")
-		configIni.setValue("SrcDom", temp);
-	//更新输入Label路径
-	temp = ui.lineInLabel->text();
-	if (temp != "")
-		configIni.setValue("SrcLabel", temp);
-	//更新输出结果路径
-	temp = ui.lineOutResult->text();
-	if (temp != "")
-		configIni.setValue("TrainResult", temp);
-
-	configIni.endGroup();//关闭组
-}
-
-void LandslideTrain::readAndStart()
-{
-	QString inDataset = ui.lineInDataset->text();
-	QString inLabel=ui.lineInLabel->text();
-	QString inModel=ui.lineInModel->text();
-	QString outResult=ui.lineOutResult->text();
-
-	if (inDataset == "" || inLabel == "" || inModel == "" || outResult == "")
-	{
-		QMessageBox mess(QMessageBox::NoIcon, QString::fromLocal8Bit("错误"), QString::fromLocal8Bit("请检查输入输出路径"));
-		mess.setWindowFlags(Qt::Drawer);
-		int result = mess.exec();
-		return;
-	}
-	ui.progressBar->SetDoubleFormatValue(QString::fromLocal8Bit("进度"), 0);
-
-	QDir inDatasetDir(inDataset);
-	if (!inDatasetDir.exists())
-	{
-		QMessageBox mess(QMessageBox::NoIcon, QString::fromLocal8Bit("错误"), QString::fromLocal8Bit("训练数据文件夹不存在"));
-		mess.setWindowFlags(Qt::Drawer);
-		int result = mess.exec();
-		return;
-	}
-	QDir inLabelDir(inLabel);
-	if (!inLabelDir.exists())
-	{
-		QMessageBox mess(QMessageBox::NoIcon, QString::fromLocal8Bit("错误"), QString::fromLocal8Bit("标签数据文件夹不存在"));
-		mess.setWindowFlags(Qt::Drawer);
-		int result = mess.exec();
-		return;
-	}
-	QDir outModelDir(outResult);
-	if (!outModelDir.exists())
-	{
-		QMessageBox mess(QMessageBox::NoIcon, QString::fromLocal8Bit("错误"), QString::fromLocal8Bit("模型输出文件夹不存在"));
-		mess.setWindowFlags(Qt::Drawer);
-		int result = mess.exec();
-		return;
-	}
-
-	if (executeGenData && executeTrain)
-		qDebug() << "executeGenData & executeTrain";
-	if (executeGenData && (!executeTrain))
-		qDebug() << "executeGenData, not executeTrain";
-	if ((!executeGenData) && executeTrain)
-	{
-		//判断train文件夹下有无训练集文件夹
-		QDir dirImage(outResult + "/Images");
-		QDir dirLabel(outResult + "/Labels");
-		if (!dirImage.exists() || !dirLabel.exists())
-		{
-			QMessageBox mess(QMessageBox::NoIcon, QString::fromLocal8Bit("错误"),
-				QString::fromLocal8Bit("未找到训练集文件夹\n确保Images和Labels在以下路径中:      \n") + outResult);
-			mess.setWindowFlags(Qt::Drawer);
-			int result = mess.exec();
-			return;
-		}
-		//判断文件是否匹配
-		QStringList imgList = getAllFiles(outResult + "/Images", "tif");
-		QStringList labelList = getAllFiles(outResult + "/Labels", "tif");
-		if (imgList.size() == 0 || labelList.size() == 0)
-		{
-			QMessageBox mess(QMessageBox::NoIcon, QString::fromLocal8Bit("错误"),
-				QString::fromLocal8Bit("训练集文件夹内为空"));
-			mess.setWindowFlags(Qt::Drawer);
-			int result = mess.exec();
-			return;
-		}
-		if (imgList.size() != labelList.size())
-		{
-			QMessageBox mess(QMessageBox::NoIcon, QString::fromLocal8Bit("错误"),
-				QString::fromLocal8Bit("训练集Images、Labels文件夹内文件不匹配"));
-			mess.setWindowFlags(Qt::Drawer);
-			int result = mess.exec();
-			return;
-		}
-	}
-
-	if (ui.lineEpoch1->text() == "" || ui.lineEpoch2->text() == "")
-	{
-		QMessageBox mess(QMessageBox::NoIcon, QString::fromLocal8Bit("错误"), QString::fromLocal8Bit("请输入正确的训练轮数"));
-		mess.setWindowFlags(Qt::Drawer);
-		int result = mess.exec();
-		return;
-	}
-	QString epoch1 = ui.lineEpoch1->text();
-	QString epoch2 = ui.lineEpoch2->text();
-
-	if (mWorkThread == nullptr)
-	{
-		qDebug() << "--startThread";
-		startWorkThread();
-	}
-
-	emit start(inModel, inDataset, inLabel, outResult, executeGenData, executeTrain, epoch1, epoch2);
-}
-
-QStringList LandslideTrain::getAllFiles(QString path, QString fileType)
-{
-	QDir dir(path);
-	if (!dir.exists())
-		return QStringList();
-	dir.setFilter(QDir::Files | QDir::NoSymLinks);
-	QFileInfoList list = dir.entryInfoList();
-
-	int file_count = list.count();
-	if (file_count <= 0)
-		return QStringList();
-	QStringList files;
-	for (int i = 0; i < file_count; i++)
-	{
-		QFileInfo file_info = list.at(i);
-		QString suffix = file_info.suffix();
-		if (QString::compare(suffix, QString(fileType), Qt::CaseInsensitive) == 0)
-		{
-			QString absolute_file_path = file_info.absoluteFilePath();
-			files.append(absolute_file_path);
-		}
-	}
-	return files;
-}
-
-void LandslideTrain::finished()
-{
-	QMessageBox mess(QMessageBox::NoIcon,
-		QString::fromLocal8Bit("运行结束"),
-		QString::fromLocal8Bit("结果文件生成路径\n") + ui.lineOutResult->text(),
-		QMessageBox::Ok, NULL);
-	mess.setWindowFlags(Qt::Drawer);
-	mess.setButtonText(QMessageBox::Ok, QString::fromLocal8Bit("确认"));
-	int result = mess.exec();
-
-	pbCancel();
-}
-
-void LandslideTrain::pbCancel()
-{
-	//qDebug() << "--pbtCancel";
-	delete myWidget;//调起&QDialog::destroyed
-}
-
-void LandslideTrain::chooseInModel()
-{
-	QString dirModel = QFileDialog::getOpenFileName(ui.pbtInModel, QString::fromLocal8Bit("选择输入初始训练模型文件"), "", "*.pth");
-	if (dirModel != "")
-		ui.lineInModel->setText(dirModel);
-}
-
-void LandslideTrain::chooseInDataset()
-{
-	QString dirDataset = QFileDialog::getExistingDirectory(ui.pbtInDataset, QString::fromLocal8Bit("选择输入训练数据路径"), "");
-	if (dirDataset != "")
-		ui.lineInDataset->setText(dirDataset);
-}
-
-void LandslideTrain::chooseInLabel()
-{
-	QString dirLabel = QFileDialog::getExistingDirectory(ui.pbtInLabel, QString::fromLocal8Bit("选择输入标签数据路径"), "");
-	if (dirLabel != "")
-		ui.lineInLabel->setText(dirLabel);
-}
-
-void LandslideTrain::chooseResultPath()
-{
-	QString dirResult = QFileDialog::getExistingDirectory(ui.pbtOutResult, QString::fromLocal8Bit("选择输出模型文件路径"), "");
-	if (dirResult != "")
-		ui.lineOutResult->setText(dirResult);
-}
-
-void WorkObject::runTrainWork(QString inModel, QString dataset, QString label, QString outModel, bool gen, bool train, QString epoch1, QString epoch2)
-{
-	QDir pluginsDir = QDir(qApp->applicationDirPath());
-	if (!pluginsDir.cd("models\\envs"))
-	{
-		qDebug() << "no folder models\\envs";
-		return;
-	}
-	QString exeDirName = pluginsDir.absoluteFilePath("train_3c_landslide.exe");
-
-	QString inDom = " --dom_path " + dataset + "/";
-	QString inLabel = " --label_path " + label + "/";
-	QString trainedModel = " --retrained_model " + inModel;
-	QString save_model = " --save_model " + outModel + "/";
-	QString strMid, strTrain;
-	if (gen)
-		strMid = " --exe_mid True ";
-	else
-		strMid = " --exe_mid False ";
-	if (train)
-		strTrain = " --exe_train True";
-	else
-		strTrain = " --exe_train False";
-	QString strEpoch1, strEpoch2;
-	strEpoch1 = " --epoch1 " + epoch1;
-	strEpoch2 = " --epoch2 " + epoch2;
-
-	QString ss = exeDirName + inDom + inLabel + trainedModel + save_model + strMid + strTrain + strEpoch1 + strEpoch2;
-	qDebug() << ss;
-	QProcess* pProces = new QProcess(this);
-	connect(pProces, SIGNAL(readyReadStandardOutput()), this, SLOT(on_read()));
-
-	pProces->start(ss);
-}
-
-void WorkObject::on_read()
-{
-	mProcess = (QProcess*)sender();
-	QString output = QString::fromLocal8Bit(mProcess->readAllStandardOutput());
-	if (output.toDouble() > 0)
-	{
-		qDebug() << "exe out:" << output.toDouble();
-		emit progress(output.toDouble());
-		if (output.toDouble() == 100.0)
-		{
-			delete mProcess;
-			mProcess = nullptr;
-			emit trainFinished();
-		}
-	}
-	else
-		qDebug() << "Unresolved exe out:" << output;
-}
-
-void WorkObject::on_cancel()
-{
-	if (mProcess == nullptr)
-	{
-		qDebug() << "--mProcess null";
-	}
-	else
-	{
-		QString KillStr = "taskkill /f /im train_3c_landslide.exe";
-		mProcess->startDetached(KillStr);
-		qDebug() << "--kill Process";
-	}
-}

LandslideTrain/LandslideTrain.h

@@ -1,96 +0,0 @@
-#pragma once
-
-#include <QtWidgets/QDialog>
-#include "ui_LandslideTrain.h"
-
-#include <QProcess>
-#include <QThread>
-#include <QFile>
-#include <QIcon>
-#include <QDebug>
-#include <QMessageBox>
-#include <QFileDialog>
-#include <QStringList>
-#include <QIntValidator>
-
-#include "SrsMainPluginInterFace.h"
-
-#include <string>
-#include <iostream>
-
-using namespace std;
-
-
-class WorkObject :public QObject
-{
-	Q_OBJECT
-
-public:
-	void on_cancel();
-
-public slots:
-	void runTrainWork(QString inModel, QString inDataset, QString label, QString outModel, bool gen, bool train, QString epoch1, QString epoch2);
-	void on_read();
-
-signals:
-	void progress(double value);
-	void trainFinished();
-
-private:
-	QProcess* mProcess = nullptr;
-
-};
-
-class LandslideTrain : public SrsMainInterface
-{
-    Q_OBJECT
-		Q_INTERFACES(SrsMainInterface)
-		Q_PLUGIN_METADATA(IID MainInterface_iid)
-
-public:
-    LandslideTrain();
-
-	virtual QString PannelName() override;
-	virtual QString CategoryName() override;
-
-	virtual QString EnglishName() override;
-	virtual QString ChineseName() override;
-	virtual QString Information() override;
-	virtual QString IconPath() override;
-	virtual QWidget* CenterWidget() override;
-
-	void startWorkThread();
-
-	QStringList getAllFiles(QString path, QString fileType);
-
-	//读json文件，获取历史存储的路径
-	void ReadConfigHistoryPaths(QString strPath);
-	//保存本次打开的路径到json文件
-	void WriteConfigPaths(QString strPath);
-
-public slots:
-	void readAndStart();
-	void pbCancel();
-
-	void chooseInModel();
-	void chooseInDataset();
-	void chooseInLabel();
-	void chooseResultPath();
-
-	void finished();
-
-signals:
-	void start(QString inModel, QString inDataset, QString label, QString outModel, bool gen, bool train, QString epoch1, QString epoch2);
-
-private:
-	Ui::LandslideTrainClass ui;
-
-	QDialog* myWidget = nullptr;
-
-	QThread* mWorkThread = nullptr;
-	WorkObject* mWorkObject = nullptr;
-
-	bool executeGenData = true;
-	bool executeTrain = true;
-
-};

LandslideTrain/LandslideTrain.qrc

@@ -1,6 +0,0 @@
-<RCC>
-    <qresource prefix="/LandslideTrain">
-        <file>resources/dem_vec.svg</file>
-        <file>LandslideTrain.qss</file>
-    </qresource>
-</RCC>

LandslideTrain/main.cpp

@@ -1,10 +0,0 @@
-#include "LandslideTrain.h"
-#include <QtWidgets/QApplication>
-
-int main(int argc, char *argv[])
-{
-    QApplication a(argc, argv);
-    LandslideTrain w;
-    w.CenterWidget()->show();
-    return a.exec();
-}

LandslideTrain/resources/dem_vec.svg

@@ -1,2 +0,0 @@
-<?xml version="1.0" standalone="no"?><!DOCTYPE svg PUBLIC "-//W3C//DTD SVG 1.1//EN" "http://www.w3.org/Graphics/SVG/1.1/DTD/svg11.dtd"><svg t="1650520802998" class="icon" viewBox="0 0 1132 1024" version="1.1" xmlns="http://www.w3.org/2000/svg" p-id="5053" width="28.7421875" height="26" xmlns:xlink="http://www.w3.org/1999/xlink"><defs><style type="text/css">@font-face { font-family: feedback-iconfont; src: url("//at.alicdn.com/t/font_1031158_u69w8yhxdu.woff2?t=1630033759944") format("woff2"), url("//at.alicdn.com/t/font_1031158_u69w8yhxdu.woff?t=1630033759944") format("woff"), url("//at.alicdn.com/t/font_1031158_u69w8yhxdu.ttf?t=1630033759944") format("truetype"); }
-</style></defs><path d="M1131.403629 583.018705c1.939725-47.361608-39.010016-64.495842-39.010017-64.495842l-226.408963-105.391702s0.538812-182.926804 0-232.120374a73.440128 73.440128 0 0 0-40.949741-66.651092S645.82591 35.80085 592.645128 6.974387c-27.694956-15.086747-56.5753 0-56.5753 0S367.529315 89.035513 315.480039 112.312208c-47.577133 21.229208-46.876677 66.651092-46.876677 66.651092V411.891892S120.160551 482.476314 57.873839 509.524696C-1.71881 535.333809 0.005389 589.592216 0.005389 589.592216s0.538812 185.890272 0 234.275623c-0.538812 52.264801 42.350653 68.806341 42.350654 68.806342s143.862906 80.06752 212.669247 114.713156c50.91777 25.539707 83.946969 0.646575 83.946969 0.646575l226.085675-122.525936s178.777948 92.67573 220.859195 121.879361c48.816402 33.729655 82.007245 5.65753 82.007245 5.65753s199.198938-111.156994 230.288412-126.836435c33.298605-16.703184 33.298605-58.030094 33.298605-58.030094s-2.047487-197.905788-0.107762-245.159633zM299.85448 675.371147l-206.903955-99.895816 206.903955-99.949696 210.783404 98.925953-210.783404 100.919559z m232.282018 148.335049l-199.091175 109.594438v-195.696657l199.091175-94.561573v180.663792z m32.705911-537.357589L340.911983 181.010787l223.930426-108.840101 232.766949 108.840101-232.766949 105.391702z m35.66938 64.38808l197.043688-94.561573v159.057415L600.511789 512.110995V350.790568z m235.137723 324.63446l-205.82633-99.895816 197.151451-99.949696 214.662853 99.949696-205.987974 99.895816z m227.378825 154.800797l-195.211725 103.12869v-195.696657l195.211725-94.561573v187.12954z" p-id="5054" fill="#2c2c2c"></path></svg>