mirror of
https://github.com/azaion/autopilot.git
synced 2026-04-23 03:06:34 +00:00
Tuomas Järvinen
22 lines
541 B
C++
22 lines
541 B
C++
#pragma once
|
|
|
|
#include <QObject>
|
|
#include "aiengineinference.h"
|
|
#include "yolov8Predictor.h"
|
|
|
|
class AiEngineInferencevOnnxRuntime : public AiEngineInference
|
|
{
|
|
Q_OBJECT
|
|
public:
|
|
explicit AiEngineInferencevOnnxRuntime(QString modelPath, QObject *parent = nullptr);
|
|
void initialize(int number);
|
|
|
|
public slots:
|
|
void performInferenceSlot(cv::Mat frame) override;
|
|
|
|
private:
|
|
cv::Mat drawLabels(const cv::Mat &image, const std::vector<Yolov8Result> &detections);
|
|
YOLOPredictor mPredictor;
|
|
QVector<QString> mClassNames;
|
|
};
|