43 lines
821 B
C
43 lines
821 B
C
|
#ifndef PERSIONMODELDETECTOR_H
|
||
|
|
#define PERSIONMODELDETECTOR_H
|
||
|
|
|
||
|
|
#include <memory>
|
||
|
|
|
||
|
|
#include <QImage>
|
||
|
|
#include <QRectF>
|
||
|
|
#include <QString>
|
||
|
|
#include <QVector>
|
||
|
|
|
||
|
|
class PersionModelDetector
|
||
|
|
{
|
||
|
|
public:
|
||
|
|
struct Detection
|
||
|
|
{
|
||
|
|
int classId = -1;
|
||
|
|
QString label;
|
||
|
|
double confidence = 0.0;
|
||
|
|
QRectF boundingBox;
|
||
|
|
};
|
||
|
|
|
||
|
|
struct Analysis
|
||
|
|
{
|
||
|
|
QVector<Detection> detections;
|
||
|
|
};
|
||
|
|
|
||
|
|
PersionModelDetector();
|
||
|
|
~PersionModelDetector();
|
||
|
|
|
||
|
|
PersionModelDetector(const PersionModelDetector&) = delete;
|
||
|
|
PersionModelDetector& operator=(const PersionModelDetector&) = delete;
|
||
|
|
|
||
|
|
bool Analyze(const QImage& frame,
|
||
|
|
Analysis& result,
|
||
|
|
QString& errorMessage);
|
||
|
|
|
||
|
|
private:
|
||
|
|
class Impl;
|
||
|
|
std::unique_ptr<Impl> m_impl;
|
||
|
|
};
|
||
|
|
|
||
|
|
#endif // PERSIONMODELDETECTOR_H
|