Hu Chunming / FFNvDecoder_WIN

#pragma once

#ifndef RESULT_H
#define RESULT_H

#include <fstream>
#include <iterator>
#include <memory>
#include <sstream>
#include <string>
#include <vector>


struct DogPoseResult {
	int x;
	int y;
	int height;
	int width;
	float confidence;
	int classId;
	std::string className;
};

// @brief 处理yolov5的结果
 class ResultYolov5 {
public:
	std::vector<std::string> class_names;
	float factor;

	//ResultYolov5();
	void read_class_names(std::string path_name);
	std::vector<DogPoseResult> yolov5_result(float* result, float threshold);
};


#endif // !RESULT_H