387 lines
11 KiB
C++
387 lines
11 KiB
C++
#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ģ<EFBFBD><EFBFBD>");
|
||
}
|
||
|
||
QString ModelAssessment::EnglishName()
|
||
{
|
||
return QString::fromLocal8Bit("DEM_Module");
|
||
}
|
||
|
||
QString ModelAssessment::ChineseName()
|
||
{
|
||
return QString::fromLocal8Bit("<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>");
|
||
}
|
||
|
||
QString ModelAssessment::Information()
|
||
{
|
||
return "<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>";
|
||
}
|
||
|
||
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";
|
||
//<2F><><EFBFBD><EFBFBD>ڲ<EFBFBD><DAB2><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>
|
||
myWidget->activateWindow();
|
||
myWidget->raise();
|
||
return myWidget;
|
||
}
|
||
ui.setupUi(myWidget);
|
||
|
||
myWidget->setWindowTitle(QString::fromLocal8Bit("<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>"));
|
||
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();//<2F><><EFBFBD><EFBFBD>wait<69><74><EFBFBD>ȵ<EFBFBD><C8B5><EFBFBD>finished<65>źŶ<C5BA>Ӧ<EFBFBD>IJۺ<C4B2><DBBA><EFBFBD><EFBFBD><EFBFBD>ִ<EFBFBD><D6B4><EFBFBD><EFBFBD><EFBFBD>ɺ<EFBFBD><C9BA><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>
|
||
//mWorkThread = nullptr;//mWorkThread, &QThread::finished, mWorkThread, &QObject::deleteLater<65><72><EFBFBD><EFBFBD><EFBFBD><EFBFBD>Ҫdelete
|
||
//mWorker = nullptr;//mWorkThread, &QThread::finished, mWorker, &QObject::deleteLater
|
||
mWorkThread->requestInterruption();//<2F><><EFBFBD><EFBFBD><EFBFBD>߳<EFBFBD><DFB3>ж<EFBFBD>
|
||
mWorkThread->quit();
|
||
mWorkThread->wait();//<2F><><EFBFBD><EFBFBD>wait<69><74><EFBFBD>ȵ<EFBFBD><C8B5><EFBFBD>finished<65>źŶ<C5BA>Ӧ<EFBFBD>IJۺ<C4B2><DBBA><EFBFBD><EFBFBD><EFBFBD>ִ<EFBFBD><D6B4><EFBFBD><EFBFBD><EFBFBD>ɺ<EFBFBD><C9BA><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>
|
||
mWorker->on_cancel();
|
||
mWorkThread = nullptr;//mWorkThread, &QThread::finished, mWorkThread, &QObject::deleteLater<65><72><EFBFBD><EFBFBD><EFBFBD><EFBFBD>Ҫ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); //<2F><>ֻ<EFBFBD><D6BB><EFBFBD><EFBFBD>ʽ<EFBFBD><CABD><EFBFBD><EFBFBD>
|
||
if (qssFile.isOpen())
|
||
{
|
||
QString qss = QLatin1String(qssFile.readAll());
|
||
myWidget->setStyleSheet(qss);
|
||
qssFile.close();
|
||
}
|
||
else
|
||
qDebug() << "<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ļ<EFBFBD>";
|
||
|
||
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 <20>ٻ<EFBFBD><D9BB><EFBFBD>
|
||
// precision <20><>ȷ<EFBFBD><C8B7>
|
||
// mIoU ƽ<><C6BD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>
|
||
// mAP ƽ<><C6BD><EFBFBD><EFBFBD><EFBFBD>Ⱦ<EFBFBD>ֵ
|
||
// 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("ƽ<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>Ⱦ<EFBFBD>ֵmAP"));
|
||
line.replace(QStringLiteral("mIoU"), QStringLiteral("ƽ<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>mIoU"));
|
||
outLines = outLines + line + "\n";
|
||
}
|
||
else if (line.contains("===>recall"))
|
||
{
|
||
//===>recall:66.62; precision: 51.2
|
||
line.remove("===>");
|
||
line.replace(QStringLiteral("recall"), QStringLiteral("<EFBFBD>ٻ<EFBFBD><EFBFBD><EFBFBD>recall"));
|
||
line.replace(QStringLiteral("precision"), QStringLiteral("<EFBFBD><EFBFBD>ȷ<EFBFBD><EFBFBD>precision"));
|
||
outLines = outLines + line + "\n";
|
||
}
|
||
else if (line.contains("===>f1_score"))
|
||
{
|
||
//===>f1_score:57.9
|
||
line.remove("===>");
|
||
line.replace(QStringLiteral("f1_score"), QStringLiteral("F1<EFBFBD><EFBFBD><EFBFBD><EFBFBD>f1_score"));
|
||
outLines = outLines + line;
|
||
}
|
||
}
|
||
file.close();
|
||
}
|
||
|
||
QMessageBox mess(QMessageBox::Information,
|
||
QString::fromLocal8Bit("<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>"),
|
||
QString::fromLocal8Bit("<EFBFBD><EFBFBD><EFBFBD>۽<EFBFBD><EFBFBD><EFBFBD>:\n") + txtDir + "\n"
|
||
+ outLines);
|
||
mess.setWindowFlags(Qt::Drawer);
|
||
mess.setTextInteractionFlags(Qt::TextSelectableByMouse);
|
||
mess.setStandardButtons(QMessageBox::Yes);
|
||
mess.button(QMessageBox::StandardButton::Yes)->setText(QString::fromLocal8Bit("ȷ<EFBFBD><EFBFBD>"));
|
||
//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("ѡ<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ģ<EFBFBD><EFBFBD><EFBFBD>ļ<EFBFBD>"), "", "*.pth");
|
||
if (fileModel != "")
|
||
ui.lineModel->setText(fileModel);
|
||
}
|
||
void ModelAssessment::choosePredictPath()
|
||
{
|
||
QString dirDom = QFileDialog::getExistingDirectory(ui.pbtInPredict, QString::fromLocal8Bit("ѡ<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ļ<EFBFBD>·<EFBFBD><EFBFBD>"), "");
|
||
if (dirDom != "")
|
||
ui.lineInPredict->setText(dirDom);
|
||
}
|
||
void ModelAssessment::chooseLabelPath()
|
||
{
|
||
QString dirDom = QFileDialog::getExistingDirectory(ui.pbtInLabel, QString::fromLocal8Bit("ѡ<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>Label<EFBFBD>ļ<EFBFBD>·<EFBFBD><EFBFBD>"), "");
|
||
if (dirDom != "")
|
||
ui.lineInLabel->setText(dirDom);
|
||
}
|
||
void ModelAssessment::chooseAssessFile()
|
||
{
|
||
QString dirDom = QFileDialog::getOpenFileName(ui.pbtIndex, QString::fromLocal8Bit("ѡ<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ļ<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>"), "", "*.txt");
|
||
if (dirDom != "")
|
||
ui.lineIndex->setText(dirDom);
|
||
}
|
||
void ModelAssessment::chooseResultPath()
|
||
{
|
||
QString dirResult = QFileDialog::getExistingDirectory(ui.pbtOutResult, QString::fromLocal8Bit("ѡ<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>·<EFBFBD><EFBFBD>"), "");
|
||
if (dirResult != "")
|
||
ui.lineOutResult->setText(dirResult);
|
||
}
|
||
|
||
void ModelAssessment::ReadConfigHistoryPaths(QString strPath)
|
||
{
|
||
QSettings configIni(strPath, QSettings::IniFormat);
|
||
|
||
//<2F><EFBFBD><F2BFAAB1><EFBFBD>Ϊ<EFBFBD><CEAA>[DemTrain] <20><><EFBFBD>飬<EFBFBD><E9A3AC>ȡDemTrain<69><6E><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>TrainResult<6C>ֶ<EFBFBD>
|
||
configIni.beginGroup("DemTrain");
|
||
|
||
QString strDemTrainResult = configIni.value("TrainResult").toString();
|
||
QDir srcDir = QDir(strDemTrainResult);
|
||
//<2F><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ļ<EFBFBD><C4BC>б<EFBFBD>
|
||
if (srcDir.absoluteFilePath("val_list.txt") != "")
|
||
ui.lineIndex->setText(srcDir.absoluteFilePath("val_list.txt"));
|
||
//<2F><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>Images
|
||
srcDir = QDir(strDemTrainResult);
|
||
if (srcDir.cd("Images"))
|
||
ui.lineInPredict->setText(srcDir.absolutePath());
|
||
//<2F><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>Labels
|
||
srcDir = QDir(strDemTrainResult);
|
||
if (srcDir.cd("Labels"))
|
||
ui.lineInLabel->setText(srcDir.absolutePath());
|
||
|
||
configIni.endGroup();//<2F>ر<EFBFBD><D8B1><EFBFBD>
|
||
|
||
//<2F><EFBFBD><F2BFAAB1><EFBFBD>Ϊ<EFBFBD><CEAA>[DemAssess] <20><><EFBFBD>飬<EFBFBD><E9A3AC>ȡAssessResult<6C>ֶ<EFBFBD>
|
||
configIni.beginGroup("DemAssess");
|
||
//<2F><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>۽<EFBFBD><DBBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>
|
||
ui.lineOutResult->setText(configIni.value("AssessResult").toString());
|
||
|
||
configIni.endGroup();//<2F>ر<EFBFBD><D8B1><EFBFBD>
|
||
}
|
||
|
||
void ModelAssessment::WriteConfigPaths(QString strPath)
|
||
{
|
||
QSettings configIni(strPath, QSettings::IniFormat);
|
||
configIni.setIniCodec("utf-8");
|
||
//<2F><EFBFBD><F2BFAAB1><EFBFBD>Ϊ<EFBFBD><CEAA>[DemAssess] <20><><EFBFBD><EFBFBD>
|
||
configIni.beginGroup("DemAssess");
|
||
|
||
//<2F><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ģ<EFBFBD><C4A3>·<EFBFBD><C2B7>
|
||
QString temp = ui.lineOutResult->text();
|
||
if (temp != "")
|
||
configIni.setValue("AssessResult", temp);
|
||
|
||
configIni.endGroup();//<2F>ر<EFBFBD><D8B1><EFBFBD>
|
||
}
|
||
|
||
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("<EFBFBD><EFBFBD><EFBFBD><EFBFBD>"), QString::fromLocal8Bit("<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>·<EFBFBD><EFBFBD>"));
|
||
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;//<2F><><EFBFBD><EFBFBD>&QDialog::destroyed
|
||
}
|