ObjectDetector.hpp

#pragma once
 
#include <cstdint>
#include <string>
#include <vector>
 
#include <opencv2/core/mat.hpp>
#include <opencv2/dnn/dnn.hpp>
 
#include "helpers/Globals.hpp"
#include "helpers/Information.hpp"

namespace LaneAndObjectDetection
{
    class ObjectDetector
    {
    public:
        explicit ObjectDetector();

        void SetProperties(const std::string& p_yoloFolderPath,
                           const Globals::ObjectDetectorTypes& p_objectDetectorTypes,
                           const Globals::ObjectDetectorBackEnds& p_objectDetectorBackEnds,
                           const Globals::ObjectDetectorBlobSizes& p_objectDetectorBlobSizes);

        void RunObjectDetector(const cv::Mat& p_frame);

        ObjectDetectionInformation GetInformation();
 
    private:
        cv::dnn::Net m_net;

        ObjectDetectionInformation m_objectDetectionInformation;

        std::vector<std::string> m_unconnectedOutputLayerNames;

        int32_t m_blobSize;

        bool m_skipObjectDetection;
    };
}