apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2025-10-06 17:17:14 +08:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Update ppseg with eigen functions (#238)

Browse Source 
* Update ppseg backend support type

* Update ppseg preprocess if condition

* Update README.md

* Update README.md

* Update README.md

* Update ppseg with eigen functions

* Delete old argmax function

* Update README.md

* Delete apply_softmax in ppseg example demo

* Update ppseg code with createFromTensor function

* Delete FDTensor2CVMat function

* Update README.md

* Update README.md

* Update README.md

* Update README.md

* Update ppseg model.cc with transpose&&softmax in place convert

* Update segmentation_result.md

* Update model.cc

* Update README.md

* Update README.md

Co-authored-by: Jason <jiangjiajun@baidu.com>
This commit is contained in:
huangjianhui
2022-09-22 21:21:47 +08:00
committed by GitHub
parent 5ad7f64a3a
commit 625845c7d6
13 changed files with 237 additions and 204 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
fastdeploy/vision/segmentation/ppseg/model.h
View File

@@ -18,7 +18,7 @@ class FASTDEPLOY_DECL PaddleSegModel : public FastDeployModel {
virtual bool Predict(cv::Mat* im, SegmentationResult* result);
bool with_softmax = false;
bool apply_softmax = false;
bool is_vertical_screen = false;
@@ -27,20 +27,21 @@ class FASTDEPLOY_DECL PaddleSegModel : public FastDeployModel {
bool BuildPreprocessPipelineFromConfig();
bool Preprocess(Mat* mat, FDTensor* outputs,
std::map<std::string, std::array<int, 2>>* im_info);
bool Preprocess(Mat* mat, FDTensor* outputs);
bool Postprocess(FDTensor& infer_result, SegmentationResult* result,
std::map<std::string, std::array<int, 2>>* im_info);
bool Postprocess(FDTensor* infer_result, SegmentationResult* result,
const std::map<std::string, std::array<int, 2>>& im_info);
bool is_resized = false;
bool is_with_softmax = false;
bool is_with_argmax = true;
std::vector<std::shared_ptr<Processor>> processors_;
std::string config_file_;
};
void FDTensor2FP32CVMat(cv::Mat& mat, FDTensor& infer_result,
bool contain_score_map);
} // namespace segmentation
} // namespace vision
} // namespace fastdeploy