File size: 5,277 Bytes
8207382 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 | // Copyright (c) 2025 PaddlePaddle Authors. All Rights Reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
#include "args.h"
DEFINE_string(input, "",
"Data to be predicted, required. Local path of an image file.");
DEFINE_string(save_path, "./output/", "Path to save inference result files.");
DEFINE_string(doc_orientation_classify_model_name, "PP-LCNet_x1_0_doc_ori",
"Name of the document image orientation classification model.");
DEFINE_string(
doc_orientation_classify_model_dir, "",
"Path to the document image orientation classification model directory.");
DEFINE_string(doc_unwarping_model_name, "UVDoc",
"Name of the text image unwarping model.");
DEFINE_string(doc_unwarping_model_dir, "",
"Path to the image unwarping model directory.");
DEFINE_string(text_detection_model_name, "PP-OCRv5_server_det",
"Name of the text detection model.");
DEFINE_string(text_detection_model_dir, "",
"Path to the text detection model directory.");
DEFINE_string(textline_orientation_model_name, "PP-LCNet_x1_0_textline_ori",
"Name of the text line orientation classification model.");
DEFINE_string(
textline_orientation_model_dir, "",
"Path to the text line orientation classification model directory.");
DEFINE_string(textline_orientation_batch_size, "6",
"Batch size for the text line orientation classification model.");
DEFINE_string(text_recognition_model_name, "PP-OCRv5_server_rec",
"Name of the text recognition model.");
DEFINE_string(text_recognition_model_dir, "",
"Path to the text recognition model directory.");
DEFINE_string(text_recognition_batch_size, "6",
"Batch size for the text recognition model.");
DEFINE_string(use_doc_orientation_classify, "true",
"Whether to use document image orientation classification.");
DEFINE_string(use_doc_unwarping, "true",
"Whether to use text image unwarping.");
DEFINE_string(use_textline_orientation, "true",
"Whether to use text line orientation classification.");
DEFINE_string(text_det_limit_side_len, "64",
"This sets a limit on the side length of the input image for the "
"text detection model.");
DEFINE_string(text_det_limit_type, "min",
"This determines how the side length limit is applied to the "
"input image before feeding it into the text detection model.");
DEFINE_string(text_det_thresh, "0.3",
"Detection pixel threshold for the text detection model. Pixels "
"with scores greater than this threshold in the output "
"probability map are considered text pixels.");
DEFINE_string(
text_det_box_thresh, "0.6",
"Detection box threshold for the text detection model. A detection result "
"is considered a text region if the average score of all pixels within the "
"border of the result is greater than this threshold.");
DEFINE_string(
text_det_unclip_ratio, "1.5",
"Text detection expansion coefficient, which expands the text region using "
"this method. The larger the value, the larger the expansion area.");
DEFINE_string(text_det_input_shape, "",
"Input shape of the text detection model.eg C,H,W");
DEFINE_string(text_rec_score_thresh, "0",
"Text recognition threshold. Text results with scores greater "
"than this threshold are retained.");
DEFINE_string(text_rec_input_shape, "",
"Input shape of the text recognition model.eg C,H,W");
DEFINE_string(lang, "", "Language in the input image for OCR processing.");
DEFINE_string(ocr_version, "", "PP-OCR version to use.");
#ifdef WITH_GPU
DEFINE_string(device, "gpu:0",
"Device for inference. Supports specifying a specific card "
"number: gpu:0.");
#else
DEFINE_string(device, "cpu",
"Device for inference. Supports specifying a specific card "
"number: gpu:0.");
#endif
DEFINE_string(vis_font_dir, "",
"When enable USE_FREETYPE, required. Path to the visualization "
"font, render the detected texts on images");
DEFINE_string(precision, "fp32",
"Computational precision, such as fp32, fp16.");
DEFINE_string(enable_mkldnn, "true", "enable_mkldnn");
DEFINE_string(mkldnn_cache_capacity, "10", "MKL-DNN cache capacity.");
DEFINE_string(cpu_threads, "8",
"Number of threads used for paddlepaddle inference on CPU.");
DEFINE_string(thread_num, "1",
"Number of threads used for pipeline instance inference on CPU.");
DEFINE_string(paddlex_config, "",
"Path to the PaddleX pipeline configuration file.");
|