azaion/autopilot
1
0
Fork 0
mirror of https://github.com/azaion/autopilot.git synced 2026-04-22 16:16:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fully working YOLOv8 detection with ONNX runtime.

Browse Source
This commit is contained in:
Tuomas Järvinen
2024-07-14 18:08:05 +02:00
parent 78cfe484b1
commit 813251b170
13 changed files with 678 additions and 49 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tmp/opi_rtsp/aiengine.cpp
+8 -4
View File
@@ -3,10 +3,12 @@
#include "aiengine.h"
#include "aiengineinference.h"
#ifdef OPI5_BUILD
#if defined(OPI5_BUILD)
#include "src-opi5/aiengineinferenceopi5.h"
#else
#elif defined(OPENCV_BUILD)
#include "src-opencv-onnx/aiengineinferenceopencvonnx.h"
#else
#include "src-onnx-runtime/aiengineinferenceonnxruntime.h"
#endif
AiEngine::AiEngine(QString modelPath, QObject *parent)
@@ -15,10 +17,12 @@ AiEngine::AiEngine(QString modelPath, QObject *parent)
mRtspListener = new AiEngineRtspListener(this);
connect(mRtspListener, &AiEngineRtspListener::frameReceived, this, &AiEngine::frameReceivedSlot);
#ifdef OPI5_BUILD
#if defined(OPI5_BUILD)
mInference = new AiEngineInferenceOpi5(modelPath);
#else
#elif defined(OPENCV_BUILD)
mInference = new AiEngineInferenceOpencvOnnx(modelPath);
#else
mInference = new AiEngineInferencevOnnxRuntime(modelPath);
#endif
QThread *inferenceThread = new QThread(this);