Renamed opi_player as rtsp_ai_player

This commit is contained in:
Tuomas Järvinen
2024-07-14 19:52:45 +02:00
parent 813251b170
commit 8a7d681e5b
40 changed files with 43 additions and 0 deletions
@@ -0,0 +1,20 @@
#pragma once
#include <QObject>
#include "aiengineinference.h"
#include "yolov8Predictor.h"
class AiEngineInferencevOnnxRuntime : public AiEngineInference
{
Q_OBJECT
public:
explicit AiEngineInferencevOnnxRuntime(QString modelPath, QObject *parent = nullptr);
public slots:
void performInferenceSlot(cv::Mat frame) override;
private:
cv::Mat drawLabels(const cv::Mat &image, const std::vector<Yolov8Result> &detections);
YOLOPredictor mPredictor;
std::vector<std::string> mClassNames;
};