Fix bug for part of detection model (#243)

* fix error for part of detection model

* fix error for part of detection model

* add patch paddle inference
This commit is contained in:
Jason
2022-09-16 16:31:34 +08:00
committed by GitHub
parent 4d2fbcb030
commit 5046574b95
10 changed files with 61 additions and 3 deletions

View File

@@ -157,6 +157,8 @@ bool PPYOLOE::Preprocess(Mat* mat, std::vector<FDTensor>* outputs) {
return false;
}
}
Cast::Run(mat, "float");
outputs->resize(2);
(*outputs)[0].name = InputInfoOfRuntime(0).name;