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
|
#pragma once
#include <fstream>
#include <iostream>
#include <sstream>
#include <vector>
#include "NvInfer.h"
#include "NvOnnxParser.h"
#include "AlgorithmResult.h"
class DogPoseDetector {
public:
bool init();
bool detect(unsigned char *pGpuBgb, int src_width, int src_height);
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 };
};
|