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

[Benchmark] Add GetModelResoucesNameFromDir APIs (#1640)

Browse Source
This commit is contained in:
DefTruth
2023-03-17 17:05:24 +08:00
committed by GitHub
parent 4d57b62f83
commit 95820a9892
14 changed files with 207 additions and 60 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
benchmark/cpp/benchmark_ppocr_rec.cc
View File

@@ -35,9 +35,14 @@ int main(int argc, char* argv[]) {
std::unordered_map<std::string, std::string> config_info;
benchmark::ResultManager::LoadBenchmarkConfig(FLAGS_config_path,
&config_info);
// Recognition Model
auto rec_model_file = FLAGS_model + sep + "inference.pdmodel";
auto rec_params_file = FLAGS_model + sep + "inference.pdiparams";
std::string model_name, params_name, config_name;
auto model_format = fastdeploy::ModelFormat::PADDLE;
if (!UpdateModelResourceName(&model_name, &params_name, &config_name,
&model_format, config_info, false)) {
return -1;
}
auto model_file = FLAGS_model + sep + model_name;
auto params_file = FLAGS_model + sep + params_name;
if (config_info["backend"] == "paddle_trt") {
option.paddle_infer_option.collect_trt_shape = true;
}
@@ -49,7 +54,7 @@ int main(int argc, char* argv[]) {
trt_shapes[2]);
}
auto model_ppocr_rec = vision::ocr::Recognizer(
rec_model_file, rec_params_file, FLAGS_rec_label_file, option);
model_file, params_file, FLAGS_rec_label_file, option, model_format);
std::string text;
float rec_score;
if (config_info["precision_compare"] == "true") {