autopilot/misc/rtsp_ai_player/src-ncnn/aiengineinferencencnn.h

#pragma once

#include <QObject>
#include <opencv2/core.hpp>
#include <ncnn/net.h>
#include "aiengineinference.h"


struct Object
{
    cv::Rect_<float> rect;
    int label;
    float prob;
};


class AiEngineInferencevNcnn : public AiEngineInference
{
    Q_OBJECT
public:
    explicit AiEngineInferencevNcnn(QString modelPath, QObject *parent = nullptr);
    void initialize(int number);

public slots:
    void performInferenceSlot(cv::Mat frame) override;

private:
    int detect_yolov8(const cv::Mat& bgr, std::vector<Object>& objects);
    ncnn::Net yolov8;
};