From 71f8e5c3091bbb887b4542a8918ecb7a3ce8ae8f Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=90=B4=E9=95=87?= <1742272207@qq.com> Date: Tue, 29 Nov 2022 12:16:26 +0800 Subject: [PATCH] modify .hfile --- FCOS/FCOSDection/FCOSDetection.h | 6 +++--- FCOS/FCOSPostprocess/FCOSDetectionPostProcess.h | 6 +++--- 2 files changed, 6 insertions(+), 6 deletions(-) diff --git a/FCOS/FCOSDection/FCOSDetection.h b/FCOS/FCOSDection/FCOSDetection.h index ed3b246..babf4f4 100644 --- a/FCOS/FCOSDection/FCOSDetection.h +++ b/FCOS/FCOSDection/FCOSDetection.h @@ -32,7 +32,7 @@ struct InitParam { }; class FCOSDetection { - public: +public: APP_ERROR Init(const InitParam &initParam); APP_ERROR DeInit(); APP_ERROR Inference(const std::vector &inputs, @@ -42,7 +42,7 @@ class FCOSDetection { std::vector> &objInfos); APP_ERROR Process(const std::string &imgPath); - protected: +protected: APP_ERROR ReadImage(const std::string &imgPath, MxBase::TensorBase &tensor); APP_ERROR Resize(const MxBase::TensorBase &inputTensor, MxBase::TensorBase &outputTensor); @@ -55,7 +55,7 @@ class FCOSDetection { const InitParam &initParam, std::map> &config); - private: +private: std::shared_ptr dvppWrapper_; // 封装DVPP基本编码、解码、扣图功能 std::shared_ptr model_; // 模型推理功能处理 diff --git a/FCOS/FCOSPostprocess/FCOSDetectionPostProcess.h b/FCOS/FCOSPostprocess/FCOSDetectionPostProcess.h index ff1b3d4..ef985f1 100644 --- a/FCOS/FCOSPostprocess/FCOSDetectionPostProcess.h +++ b/FCOS/FCOSPostprocess/FCOSDetectionPostProcess.h @@ -21,7 +21,7 @@ #include "opencv2/opencv.hpp" class FCOSPostProcess : public MxBase::ObjectPostProcessBase { - public: +public: FCOSPostProcess() = default; ~FCOSPostProcess() = default; FCOSPostProcess(const FCOSPostProcess &other); @@ -36,11 +36,11 @@ class FCOSPostProcess : public MxBase::ObjectPostProcessBase { override; FCOSPostProcess &operator=(const FCOSPostProcess &other); - protected: +protected: void PostprocessBBox(MxBase::ObjectInfo &objInfo, int imageWidth, int imageHeight, int netInWidth, int netInHeight); - private: +private: uint32_t classNum_ = 0; bool softmax_ = true; uint32_t topK_ = 1;