Hu Chunming / FFNvDecoder_WIN

Blame view

FFNvDecoder/DogPoseDetector.h 572 Bytes
Edit Raw Normal View History Permalink
07639e75   Hu Chunming   实现狗狗姿态检测 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
 
  #pragma once

  

  #include <fstream>

  #include <iostream>

  #include <sstream>

  #include <vector>

  

  #include "NvInfer.h"

  #include "NvOnnxParser.h"

  

  #include "AlgorithmResult.h"

  

  class DogPoseDetector {

  public:

  	bool init();
69bdb744   Hu Chunming   改用onnx模型 
17
 
  	std::vector<DogPoseResult> detect(unsigned char *pGpuBgb, int src_width, int src_height);
07639e75   Hu Chunming   实现狗狗姿态检测 
18
19
20
21
22
23
24
25
26
27
28
29
 
  

  private:

  	nvinfer1::IExecutionContext* context;

  

  	void** m_data_buffer;

  	int m_input_node_index;

  	nvinfer1::Dims m_input_node_dim;

  	int m_output_node_index;

  	nvinfer1::Dims m_output_node_dim;

  	

  	unsigned char* pSquareData{ nullptr };

  };