apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2026-04-24 01:29:57 +08:00
Code Issues Actions 9 Packages Projects Releases Wiki Activity

Support remove multiclass_nms to enable ppyoloe to tensorrt (#40)

Browse Source 
* Add custom operator for onnxruntime ans fix paddle backend

* Polish cmake files and runtime apis

* Remove copy libraries

* fix some issue

* fix bug

* fix bug

* Support remove multiclass_nms to enable paddledetection run tensorrt

* Support remove multiclass_nms to enable paddledetection run tensorrt

* Support remove multiclass_nms to enable paddledetection run tensorrt

* Support remove multiclass_nms to enable paddledetection run tensorrt

* add common operator multiclassnms

* fix compile problem

Co-authored-by: root <root@bjyz-sys-gpu-kongming3.bjyz.baidu.com>
This commit is contained in:
Jason
2022-07-26 11:16:01 +08:00
committed by GitHub
parent fc71d79e58
commit 17e4dc6b5e
18 changed files with 460 additions and 56 deletions
Download Patch File Download Diff File Expand all files Collapse all files
fastdeploy/vision/ppdet/ppyoloe.h
+10 -6
View File
@@ -25,8 +25,7 @@ class FASTDEPLOY_DECL PPYOLOE : public FastDeployModel {
virtual bool Preprocess(Mat* mat, std::vector<FDTensor>* outputs);
virtual bool Postprocess(std::vector<FDTensor>& infer_result,
DetectionResult* result, float conf_threshold,
float nms_threshold);
DetectionResult* result);
virtual bool Predict(cv::Mat* im, DetectionResult* result,
float conf_threshold = 0.5, float nms_threshold = 0.7);
@@ -34,10 +33,15 @@ class FASTDEPLOY_DECL PPYOLOE : public FastDeployModel {
private:
std::vector<std::shared_ptr<Processor>> processors_;
std::string config_file_;
// PaddleDetection can export model without nms
// This flag will help us to handle the different
// situation
bool has_nms_;
// configuration for nms
int64_t background_label = -1;
int64_t keep_top_k = 300;
float nms_eta = 1.0;
float nms_threshold = 0.7;
float score_threshold = 0.01;
int64_t nms_top_k = 10000;
bool normalized = true;
bool has_nms_ = false;
};
} // namespace ppdet
} // namespace vision