Changed directory structure and renamed applications

- autopilot -> drone_controller
- rtsp_ai_player -> ai_controller
- added top level qmake project file
- updated documentation
- moved small demo applications from tmp/ to misc/
This commit is contained in:
Tuomas Järvinen
2024-10-19 14:44:34 +02:00
parent 54b7dc41ca
commit 45c19baa45
94 changed files with 149 additions and 204 deletions
+50
View File
@@ -0,0 +1,50 @@
#pragma once
#include <QObject>
#include <QString>
#include <QVector>
#include <opencv2/core.hpp>
#include <opencv2/imgproc.hpp>
#include "aienginedefinitions.h"
const int INFERENCE_SQUARE_WIDTH = 640;
const int INFERENCE_SQUARE_HEIGHT = 640;
class AiEngineObject {
public:
AiEngineRectangle rectangle;
float propability;
int classId;
QString classStr;
};
class AiEngineInferenceResult {
public:
cv::Mat frame;
QVector<AiEngineObject> objects;
};
class AiEngineInference : public QObject
{
Q_OBJECT
public:
explicit AiEngineInference(QString modelPath, QObject *parent = nullptr);
bool isActive(void);
protected:
cv::Mat resizeAndPad(const cv::Mat& src);
QString mModelPath;
bool mActive;
int mNumber;
QVector<QString> mClassNames;
public slots:
virtual void performInferenceSlot(cv::Mat frame) = 0;
virtual void initialize(int number) = 0;
signals:
void resultsReady(AiEngineInferenceResult results);
};