32 lines
639 B
C
32 lines
639 B
C
|
#ifndef AAPGSMODELCLASSIFIER_H
|
||
|
|
#define AAPGSMODELCLASSIFIER_H
|
||
|
|
|
||
|
|
#include <memory>
|
||
|
|
|
||
|
|
#include <QImage>
|
||
|
|
#include <QString>
|
||
|
|
|
||
|
|
class AapgsModelClassifier
|
||
|
|
{
|
||
|
|
public:
|
||
|
|
struct Classification
|
||
|
|
{
|
||
|
|
QString modelType;
|
||
|
|
double confidence = 0.0;
|
||
|
|
};
|
||
|
|
|
||
|
|
AapgsModelClassifier();
|
||
|
|
~AapgsModelClassifier();
|
||
|
|
|
||
|
|
AapgsModelClassifier(const AapgsModelClassifier&) = delete;
|
||
|
|
AapgsModelClassifier& operator=(const AapgsModelClassifier&) = delete;
|
||
|
|
|
||
|
|
bool Classify(const QImage& frame, Classification& result, QString& errorMessage);
|
||
|
|
|
||
|
|
private:
|
||
|
|
class Impl;
|
||
|
|
std::unique_ptr<Impl> m_impl;
|
||
|
|
};
|
||
|
|
|
||
|
|
#endif // AAPGSMODELCLASSIFIER_H
|