zishuo/ascend_community_projects
1
0
Fork 0
mirror of https://github.com/Ascend/ascend_community_projects.git synced 2025-09-26 20:01:17 +08:00
Code Issues Packages Projects Releases Wiki Activity

modify .hfile

Browse Source
This commit is contained in:
吴镇
2022-11-29 12:16:26 +08:00
parent 25075ba411
commit 71f8e5c309
2 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
FCOS/FCOSDection/FCOSDetection.h
View File

@@ -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<MxBase::TensorBase> &inputs,
@@ -42,7 +42,7 @@ class FCOSDetection {
std::vector<std::vector<MxBase::ObjectInfo>> &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<std::string, std::shared_ptr<void>> &config);
private:
private:
std::shared_ptr<MxBase::DvppWrapper>
dvppWrapper_; // 封装DVPP基本编码、解码、扣图功能
std::shared_ptr<MxBase::ModelInferenceProcessor> model_; // 模型推理功能处理

6
FCOS/FCOSPostprocess/FCOSDetectionPostProcess.h
View File

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