apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2025-12-24 13:28:13 +08:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Merge branch 'PaddlePaddle:develop' into gbd_android

Browse Source
This commit is contained in:
Winter
2022-11-08 17:33:31 +08:00
committed by GitHub
parent 0df5e89f4e 5e75457fd7
commit 16dd478409
1 changed files with 4 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
fastdeploy/vision/detection/ppdet/ppyoloe.cc
View File

@@ -55,6 +55,7 @@ bool PPYOLOE::Initialize() {
FDERROR << "Failed to initialize fastdeploy backend." << std::endl;
return false;
}
reused_input_tensors.resize(2);
return true;
}
@@ -251,22 +252,19 @@ bool PPYOLOE::Postprocess(std::vector<FDTensor>& infer_result,
bool PPYOLOE::Predict(cv::Mat* im, DetectionResult* result) {
Mat mat(*im);
std::vector<FDTensor> processed_data;
if (!Preprocess(&mat, &processed_data)) {
if (!Preprocess(&mat, &reused_input_tensors)) {
FDERROR << "Failed to preprocess input data while using model:"
<< ModelName() << "." << std::endl;
return false;
}
float* tmp = static_cast<float*>(processed_data[1].Data());
std::vector<FDTensor> infer_result;
if (!Infer(processed_data, &infer_result)) {
if (!Infer()) {
FDERROR << "Failed to inference while using model:" << ModelName() << "."
<< std::endl;
return false;
}
if (!Postprocess(infer_result, result)) {
if (!Postprocess(reused_output_tensors, result)) {
FDERROR << "Failed to postprocess while using model:" << ModelName() << "."
<< std::endl;
return false;