|
|
@ -5,21 +5,21 @@
|
|
|
|
class MF_ObjectDetectBase :public MA_TRTInferAlgoBase
|
|
|
|
class MF_ObjectDetectBase :public MA_TRTInferAlgoBase
|
|
|
|
{
|
|
|
|
{
|
|
|
|
public:
|
|
|
|
public:
|
|
|
|
MF_ObjectDetectBase(const utils::InitParameter& param);
|
|
|
|
MF_ObjectDetectBase(const trtUtils::InitParameter& param);
|
|
|
|
~MF_ObjectDetectBase();
|
|
|
|
~MF_ObjectDetectBase();
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
protected:
|
|
|
|
protected:
|
|
|
|
std::vector<std::vector<utils::Box>> getObjectss() const;
|
|
|
|
std::vector<std::vector<trtUtils::Box>> getObjectss() const;
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
protected:
|
|
|
|
protected:
|
|
|
|
utils::InitParameter m_param;
|
|
|
|
trtUtils::InitParameter m_param;
|
|
|
|
nvinfer1::Dims m_output_dims;
|
|
|
|
nvinfer1::Dims m_output_dims;
|
|
|
|
int m_output_area;
|
|
|
|
int m_output_area;
|
|
|
|
int m_total_objects;
|
|
|
|
int m_total_objects;
|
|
|
|
std::vector<std::vector<utils::Box>> m_objectss; // Ä¿±ê¼ì²â¿ò
|
|
|
|
std::vector<std::vector<trtUtils::Box>> m_objectss; // Ä¿±ê¼ì²â¿ò
|
|
|
|
utils::AffineMat m_dst2src;
|
|
|
|
trtUtils::AffineMat m_dst2src;
|
|
|
|
|
|
|
|
|
|
|
|
// input
|
|
|
|
// input
|
|
|
|
unsigned char* m_input_src_device;
|
|
|
|
unsigned char* m_input_src_device;
|
|
|
|