You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
41 lines
836 B
C++
41 lines
836 B
C++
#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;
|
|
|
|
};
|
|
|