#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;
};