mirror of
https://github.com/azaion/autopilot.git
synced 2026-04-22 21:46:33 +00:00
Your Name
26 lines
710 B
C++
26 lines
710 B
C++
#pragma once
|
|
|
|
#include <QObject>
|
|
#include "aiengineinference.h"
|
|
#include "yolov8.h"
|
|
#include "image_utils.h"
|
|
|
|
class AiEngineInferenceOpi5 : public AiEngineInference
|
|
{
|
|
Q_OBJECT
|
|
public:
|
|
explicit AiEngineInferenceOpi5(QString modelPath, QObject *parent = nullptr);
|
|
~AiEngineInferenceOpi5();
|
|
|
|
public slots:
|
|
void performInferenceSlot(cv::Mat frame) override;
|
|
|
|
private:
|
|
image_buffer_t convertCV2FrameToImageBuffer(const cv::Mat& bgrFrame);
|
|
void freeImageBuffer(image_buffer_t& imgBuffer);
|
|
cv::Mat resizeToHalfAndAssigntoTopLeft640x640(const cv::Mat& inputFrame);
|
|
void drawObjects(cv::Mat& image, const object_detect_result_list& result_list);
|
|
|
|
rknn_app_context_t mRrknnAppCtx;
|
|
};
|