#pragma once
#include "MA_TRTInferAlgoBase.h"


class MF_ObjectDetectBase :public MA_TRTInferAlgoBase
{
public:
	MF_ObjectDetectBase(const utils::InitParameter& param);
	~MF_ObjectDetectBase();


protected:
	std::vector<std::vector<utils::Box>> getObjectss() const;


protected:
	utils::InitParameter m_param;
	nvinfer1::Dims m_output_dims;
	int m_output_area;
	int m_total_objects;
	std::vector<std::vector<utils::Box>> m_objectss;		// Ä¿±ê¼ì²â¿ò
	utils::AffineMat m_dst2src;

	// input
	unsigned char* m_input_src_device;
	float* m_input_resize_device;
	float* m_input_rgb_device;
	float* m_input_norm_device;
	float* m_input_hwc_device;

	// output
	float* m_output_src_device;
	float* m_output_objects_device;
	float* m_output_objects_host;
	int m_output_objects_width;
	int* m_output_idx_device;
	float* m_output_conf_device;

};