优化代码

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

@ -1,6 +1,6 @@
#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
{
public:
MF_ImageSegmentationBase(const utils::InitParameter& _param);
MF_ImageSegmentationBase(const trtUtils::InitParameter& _param);
~MF_ImageSegmentationBase();

@ -1,6 +1,6 @@
#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
m_input_src_device = nullptr;
@ -48,7 +48,7 @@ MF_ObjectDetectBase::~MF_ObjectDetectBase()
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;
}

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

Loading…
Cancel
Save