mirror of
https://github.com/azaion/autopilot.git
synced 2026-04-22 06:56:33 +00:00
Tuomas Järvinen
45c19baa45
- autopilot -> drone_controller - rtsp_ai_player -> ai_controller - added top level qmake project file - updated documentation - moved small demo applications from tmp/ to misc/
37 lines
912 B
C++
37 lines
912 B
C++
#pragma once
|
|
|
|
#include <QObject>
|
|
#include <QElapsedTimer>
|
|
#include <opencv2/core.hpp>
|
|
#include <opencv2/videoio.hpp>
|
|
#include "aienginertsplistener.h"
|
|
#include "aiengineinference.h"
|
|
#include "aienginegimbalclient.h"
|
|
|
|
class AiEngine : public QObject
|
|
{
|
|
Q_OBJECT
|
|
public:
|
|
explicit AiEngine(QString modelPath, QObject *parent = nullptr);
|
|
void start(void);
|
|
void stop(void);
|
|
|
|
public slots:
|
|
void frameReceivedSlot(cv::Mat frame);
|
|
void inferenceResultsReceivedSlot(AiEngineInferenceResult result);
|
|
|
|
signals:
|
|
void inferenceFrame(cv::Mat frame);
|
|
void inferenceFrame2(cv::Mat frame);
|
|
void inferenceFrame3(cv::Mat frame);
|
|
|
|
private:
|
|
QElapsedTimer mElapsedTimer;
|
|
uint32_t mFrameCounter = 0;
|
|
AiEngineRtspListener *mRtspListener;
|
|
AiEngineInference *mInference;
|
|
AiEngineInference *mInference2;
|
|
AiEngineInference *mInference3;
|
|
AiEngineGimbalClient *mGimbalClient;
|
|
};
|