优化代码

main
熊继淙 1 year ago
parent 2cd5976bde
commit 9dc17da6ba

@ -1,6 +1,6 @@
#include "MF_ImageSegmentationBase.h" #include "MF_ImageSegmentationBase.h"
MF_ImageSegmentationBase::MF_ImageSegmentationBase(const utils::InitParameter& _param) :MA_TRTInferAlgoBase(_param) MF_ImageSegmentationBase::MF_ImageSegmentationBase(const trtUtils::InitParameter& _param) :MA_TRTInferAlgoBase(_param)
{ {
} }

@ -3,7 +3,7 @@
class MF_ImageSegmentationBase : public MA_TRTInferAlgoBase class MF_ImageSegmentationBase : public MA_TRTInferAlgoBase
{ {
public: public:
MF_ImageSegmentationBase(const utils::InitParameter& _param); MF_ImageSegmentationBase(const trtUtils::InitParameter& _param);
~MF_ImageSegmentationBase(); ~MF_ImageSegmentationBase();

@ -1,6 +1,6 @@
#include "MF_ObjectDetectBase.h" #include "MF_ObjectDetectBase.h"
MF_ObjectDetectBase::MF_ObjectDetectBase(const utils::InitParameter& param) :MA_TRTInferAlgoBase(param) MF_ObjectDetectBase::MF_ObjectDetectBase(const trtUtils::InitParameter& param) :MA_TRTInferAlgoBase(param)
{ {
// input // input
m_input_src_device = nullptr; m_input_src_device = nullptr;
@ -48,7 +48,7 @@ MF_ObjectDetectBase::~MF_ObjectDetectBase()
delete[] m_output_objects_host; delete[] m_output_objects_host;
} }
std::vector<std::vector<utils::Box>> MF_ObjectDetectBase::getObjectss() const std::vector<std::vector<trtUtils::Box>> MF_ObjectDetectBase::getObjectss() const
{ {
return m_objectss; return m_objectss;
} }

@ -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;

Loading…
Cancel
Save