88 lines
2.0 KiB
C++
88 lines
2.0 KiB
C++
|
#pragma once
|
|||
|
|
|
|||
|
|
#include <iostream>
|
|||
|
|
#include <QObject>
|
|||
|
|
#include <QWidget>
|
|||
|
|
#include <QLabel>
|
|||
|
|
#include <QPushButton>
|
|||
|
|
#include "DataModelRegistry.hpp"
|
|||
|
|
#include "NodeDataModel.hpp"
|
|||
|
|
#include "halconcpp/HalconCpp.h"
|
|||
|
|
#include "HImageData.hpp"
|
|||
|
|
#include "HRegionData.hpp"
|
|||
|
|
#include "HImageViewWidget.hpp"
|
|||
|
|
#include "DrawShapeView.hpp"
|
|||
|
|
|
|||
|
|
using QtNodes::PortType;
|
|||
|
|
using QtNodes::PortIndex;
|
|||
|
|
using QtNodes::NodeData;
|
|||
|
|
using QtNodes::NodeDataType;
|
|||
|
|
using QtNodes::NodeDataModel;
|
|||
|
|
using QtNodes::NodeValidationState;
|
|||
|
|
using namespace HalconCpp;
|
|||
|
|
/**
|
|||
|
|
* \brief halcon 图像rgb2gray节点
|
|||
|
|
*/
|
|||
|
|
class HImageDLSegmentModel :public NodeDataModel
|
|||
|
|
{
|
|||
|
|
Q_OBJECT
|
|||
|
|
public:
|
|||
|
|
HImageDLSegmentModel();
|
|||
|
|
virtual ~HImageDLSegmentModel() {}
|
|||
|
|
|
|||
|
|
public:
|
|||
|
|
QString caption() const override
|
|||
|
|
{
|
|||
|
|
return QStringLiteral("语义分割");
|
|||
|
|
}
|
|||
|
|
QString name() const override
|
|||
|
|
{
|
|||
|
|
return QStringLiteral("语义分割");
|
|||
|
|
}
|
|||
|
|
virtual QString modelName() const
|
|||
|
|
{
|
|||
|
|
return QStringLiteral("语义分割");
|
|||
|
|
}
|
|||
|
|
unsigned int
|
|||
|
|
nPorts(PortType portType) const override;
|
|||
|
|
|
|||
|
|
NodeDataType
|
|||
|
|
dataType(PortType portType, PortIndex portIndex) const override;
|
|||
|
|
|
|||
|
|
std::shared_ptr<NodeData>
|
|||
|
|
outData(PortIndex port) override;
|
|||
|
|
|
|||
|
|
void
|
|||
|
|
setInData(std::shared_ptr<NodeData>, int) override;
|
|||
|
|
|
|||
|
|
QWidget*
|
|||
|
|
embeddedWidget() override { return btn_select_model; }
|
|||
|
|
|
|||
|
|
bool
|
|||
|
|
resizable() const override { return false; }
|
|||
|
|
NodeValidationState
|
|||
|
|
validationState() const override;
|
|||
|
|
QString
|
|||
|
|
validationMessage() const override;
|
|||
|
|
QJsonObject save() const override;
|
|||
|
|
void restore(QJsonObject const&) override;
|
|||
|
|
void readDlModel(QString modelFileName);
|
|||
|
|
protected:
|
|||
|
|
bool RunTask();
|
|||
|
|
|
|||
|
|
public slots:
|
|||
|
|
void OnNewRegionData(ShapeDataStruct _data);
|
|||
|
|
public:
|
|||
|
|
NodeValidationState modelValidationState = NodeValidationState::Warning;
|
|||
|
|
QString modelValidationError = QStringLiteral("图片输入未连接!");
|
|||
|
|
private:
|
|||
|
|
QString dl_path;
|
|||
|
|
QPushButton* btn_select_model;
|
|||
|
|
HDlModel* m_dl_model;
|
|||
|
|
HTuple image_dimensions;
|
|||
|
|
HTuple class_ids;
|
|||
|
|
HTuple valid_thres = 0.7;
|
|||
|
|
std::shared_ptr<HImageData> m_hImage;
|
|||
|
|
std::shared_ptr<HRegionData> m_result;
|
|||
|
|
};
|