autopilot/misc/rtsp_ai_player/src-opencv-onnx/aiengineinferenceopencvonnx.cpp

#include <QDebug>
#include <QThread>
#include "aiengineinferenceopencvonnx.h"


AiEngineInferenceOpencvOnnx::AiEngineInferenceOpencvOnnx(QString modelPath, QObject *parent)
    : AiEngineInference{modelPath, parent},
      mInference(modelPath.toStdString(), cv::Size(640, 640), "classes.txt")
{
    //qDebug() << "TUOMAS test mModelPath=" << mModelPath;
    //mEngine = new InferenceEngine(modelPath.toStdString());
    //mInference = new Inference(modelPath.toStdString(), cv::Size(INFERENCE_SQUARE_WIDTH, INFERENCE_SQUARE_HEIGHT), "classes.txt");
}


void AiEngineInferenceOpencvOnnx::performInferenceSlot(cv::Mat frame)
{
    try {
        //qDebug() << "performInferenceSlot() in thread: " << QThread::currentThreadId();

        mActive = true;

        cv::Mat scaledImage               = resizeAndPad(frame);
        std::vector<Detection> detections = mInference.runInference(scaledImage);
        AiEngineInferenceResult result;

        //qDebug() << "performInferenceSlot() found " << detections.size() << " objects";

        for (uint i = 0; i < detections.size(); ++i) {
            const Detection &detection = detections[i];

            // Add detected objects to the results
            AiEngineObject object;
            object.classId          = detection.class_id;
            object.propability      = detection.confidence;
            object.rectangle.top    = detection.box.y;
            object.rectangle.left   = detection.box.x;
            object.rectangle.bottom = detection.box.y + detection.box.height;
            object.rectangle.right  = detection.box.x + detection.box.width;
            result.objects.append(object);
        }

        if (result.objects.empty() == false) {
            result.frame = mInference.drawLabels(scaledImage, detections);
            emit resultsReady(result);
        }

        mActive = false;
    }
    catch (const cv::Exception& e) {
        std::cerr << "performInferenceSlot() Error: " << e.what() << std::endl;
    }
}