#pragma once

#include <fstream>
#include <iostream>
#include <sstream>
#include <vector>

#include "NvInfer.h"
#include "NvOnnxParser.h"

#include "AlgorithmResult.h"

class DogPoseDetector {
public:
	bool init();

	std::vector<DogPoseResult> 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 };
};