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

fix first check

Browse Source
This commit is contained in:
吴镇
2022-12-01 07:21:04 +08:00
parent 2bd996a441
commit cbc67f03e0
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
Retinaface/RetinafacePostProcess/RetinafacePostProcess.h
View File

@@ -26,7 +26,7 @@
#define DEFAULT_CONFIDENCE_THRESH 0.40
class RetinafacePostProcess : public MxBase::ObjectPostProcessBase {
public:
public:
RetinafacePostProcess() = default;
~RetinafacePostProcess() = default;
@@ -47,7 +47,7 @@ class RetinafacePostProcess : public MxBase::ObjectPostProcessBase {
const std::map<std::string, std::shared_ptr<void>>& paramMap = {})
override;
protected:
protected:
void ObjectDetectionOutput(
const std::vector<MxBase::TensorBase>& tensors,
std::vector<std::vector<MxBase::ObjectInfo>>& objectInfos,
@@ -56,7 +56,7 @@ class RetinafacePostProcess : public MxBase::ObjectPostProcessBase {
cv::Mat decode_for_loc(cv::Mat& loc, cv::Mat& prior, cv::Mat& key,
float resize_scale_factor);
private:
private:
uint32_t objectConfTensor_ = DEFAULT_OBJECT_CONF_TENSOR;
uint32_t objectInfoTensor_ = DEFAULT_OBJECT_INFO_TENSOR;
float iouThresh_ = DEFAULT_IOU_THRESH;