Blame view

FFNvDecoder/AlgorithmResult.h 608 Bytes
07639e75   Hu Chunming   实现狗狗姿态检测
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
  #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