From ba8649a69de5561dfdece0cf7044ce00b2dba0af Mon Sep 17 00:00:00 2001 From: DefTruth <31974251+DefTruth@users.noreply.github.com> Date: Sun, 25 Jun 2023 13:50:02 +0800 Subject: [PATCH] [Model] update PP-ShiTuV2-rec preprocess parser policy (#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 --- .../classification/ppshitu/ppshituv2_rec_preprocessor.cc | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/fastdeploy/vision/classification/ppshitu/ppshituv2_rec_preprocessor.cc b/fastdeploy/vision/classification/ppshitu/ppshituv2_rec_preprocessor.cc index acb35c61b3..aa0615a382 100644 --- 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()); for (const auto& op : preprocess_cfg) { FDASSERT(op.IsMap(),