#pragma once
#include <opencv2/opencv.hpp>
#include <opencv2/dnn.hpp>
#include <vector>
#include <string>

class YoloFeatureExtractor {
private:
    cv::dnn::Net net;
    int inputWidth;
    int inputHeight;
    std::vector<std::string> classNames;

public:
    YoloFeatureExtractor(const std::string& modelPath, const std::string& classesPath);
    ~YoloFeatureExtractor() = default;

    void loadClassNames(const std::string& file);
    std::vector<float> extractFeatures(const std::string& imagePath);
    std::vector<float> extractBackboneFeatures(const std::string& imagePath);
    std::vector<std::vector<float>> extractROIFeatures(const std::string& imagePath);

    void initOpenCL();
};