zhipuzi_pos_windows/zhipuzi_pos_windows/ai/YoloFeatureExtractor.h


#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> globalAveragePooling(const cv::Mat & featureMap);
	void normalizeL2(cv::Mat& feat);
	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);
};