[Model] update PP-ShiTuV2-rec preprocess parser policy (PaddlePaddle#…

…2061) * [benchmark] fixed paddlex benchmark for picodet 320 * [Bug Fix] fixed paddlex ppseg pp-trt infer error * [Bug Fix] fixed paddlex dino benchmark trt shapes * [benchmark] support paddlex ppyoloe pptrt benchmark * [benchmark] adjust paddlex dino trt shapes * [benchmark] add max_workspace_size flags for tensorrt/pptrt backend * [benchmark] add max_workspace_size flags for tensorrt/pptrt backend * [benchmark] add max_workspace_size flags for tensorrt/pptrt backend * [benchmark] add ort/paddle h2d gpu configs for paddlex * [benchmark] update paddlex benchmark scripts * [benchmark] update paddlex benchmark scripts * [Model] update PP-ShituV2-rec preprocess parser policy --------- Co-authored-by: qiuyanjun <[email protected]>
ArthurSec · Jun 25, 2023 · ba8649a · ba8649a
1 parent 4aff322
commit ba8649a
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/fastdeploy/vision/classification/ppshitu/ppshituv2_rec_preprocessor.cc b/fastdeploy/vision/classification/ppshitu/ppshituv2_rec_preprocessor.cc
@@ -37,7 +37,10 @@ bool PPShiTuV2RecognizerPreprocessor::BuildPreprocessPipelineFromConfig() {
             << ", maybe you should check this file." << std::endl;
     return false;
   }
-  auto preprocess_cfg = cfg["PreProcess"]["transform_ops"];
+  // auto preprocess_cfg = cfg["PreProcess"]["transform_ops"];
+  // We use the key 'RecPreProcess' to denote the preprocess 
+  // operators for PP-ShiTuV2 recognizer.
+  auto preprocess_cfg = cfg["RecPreProcess"]["transform_ops"];
   processors_.push_back(std::make_shared<BGR2RGB>());
   for (const auto& op : preprocess_cfg) {
     FDASSERT(op.IsMap(),