help="Path to the layout detection model directory.",
)
subparser.add_argument(
"--layout_threshold",
type=float,
help="Score threshold for the layout detection model.",
)
subparser.add_argument(
"--layout_nms",
type=str2bool,
help="Whether to use NMS in layout detection.",
)
subparser.add_argument(
"--layout_unclip_ratio",
type=float,
help="Expansion coefficient for layout detection.",
)
subparser.add_argument(
"--layout_merge_bboxes_mode",
type=str,
help="Overlapping box filtering method.",
)
subparser.add_argument(
"--chart_recognition_model_name",
type=str,
help="Name of the chart recognition model.",
)
subparser.add_argument(
"--chart_recognition_model_dir",
type=str,
help="Path to the chart recognition model directory.",
)
subparser.add_argument(
"--chart_recognition_batch_size",
type=int,
help="Batch size for the chart recognition model.",
)
subparser.add_argument(
"--region_detection_model_name",
type=str,
help="Name of the region detection model.",
)
subparser.add_argument(
"--region_detection_model_dir",
type=str,
help="Path to the region detection model directory.",
)
subparser.add_argument(
"--doc_orientation_classify_model_name",
type=str,
help="Name of the document image orientation classification model.",
)
subparser.add_argument(
"--doc_orientation_classify_model_dir",
type=str,
help="Path to the document image orientation classification model directory.",
)
subparser.add_argument(
"--doc_unwarping_model_name",
type=str,
help="Name of the text image unwarping model.",
)
subparser.add_argument(
"--doc_unwarping_model_dir",
type=str,
help="Path to the image unwarping model directory.",
)
subparser.add_argument(
"--text_detection_model_name",
type=str,
help="Name of the text detection model.",
)
subparser.add_argument(
"--text_detection_model_dir",
type=str,
help="Path to the text detection model directory.",
)
subparser.add_argument(
"--text_det_limit_side_len",
type=int,
help="This sets a limit on the side length of the input image for the text detection model.",
)
subparser.add_argument(
"--text_det_limit_type",
type=str,
help="This determines how the side length limit is applied to the input image before feeding it into the text deteciton model.",
)
subparser.add_argument(
"--text_det_thresh",
type=float,
help="Detection pixel threshold for the text detection model. Pixels with scores greater than this threshold in the output probability map are considered text pixels.",
)
subparser.add_argument(
"--text_det_box_thresh",
type=float,
help="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.",
)
subparser.add_argument(
"--text_det_unclip_ratio",
type=float,
help="Text detection expansion coefficient, which expands the text region using this method. The larger the value, the larger the expansion area.",
)
subparser.add_argument(
"--textline_orientation_model_name",
type=str,
help="Name of the text line orientation classification model.",
)
subparser.add_argument(
"--textline_orientation_model_dir",
type=str,
help="Path to the text line orientation classification directory.",
)
subparser.add_argument(
"--textline_orientation_batch_size",
type=int,
help="Batch size for the text line orientation classification model.",
)
subparser.add_argument(
"--text_recognition_model_name",
type=str,
help="Name of the text recognition model.",
)
subparser.add_argument(
"--text_recognition_model_dir",
type=str,
help="Path to the text recognition model directory.",
)
subparser.add_argument(
"--text_recognition_batch_size",
type=int,
help="Batch size for the text recognition model.",
)
subparser.add_argument(
"--text_rec_score_thresh",
type=float,
help="Text recognition threshold used in general OCR. Text results with scores greater than this threshold are retained.",
)
subparser.add_argument(
"--table_classification_model_name",
type=str,
help="Name of the table classification model.",
)
subparser.add_argument(
"--table_classification_model_dir",
type=str,
help="Path to the table classification model directory.",
)
subparser.add_argument(
"--wired_table_structure_recognition_model_name",
type=str,
help="Name of the wired table structure recognition model.",
)
subparser.add_argument(
"--wired_table_structure_recognition_model_dir",
type=str,
help="Path to the wired table structure recognition model directory.",
help="Path to the wired table structure recognition model directory.",
)
subparser.add_argument(
"--wired_table_cells_detection_model_name",
type=str,
help="Name of the wired table cells detection model.",
)
subparser.add_argument(
"--wired_table_cells_detection_model_dir",
type=str,
help="Path to the wired table cells detection model directory.",
)
subparser.add_argument(
"--wireless_table_cells_detection_model_name",
type=str,
help="Name of the wireless table cells detection model.",
)
subparser.add_argument(
"--wireless_table_cells_detection_model_dir",
type=str,
help="Path to the wireless table cells detection model directory.",
)
subparser.add_argument(
"--seal_text_detection_model_name",
type=str,
help="Name of the seal text detection model.",
)
subparser.add_argument(
"--seal_text_detection_model_dir",
type=str,
help="Path to the seal text detection model directory.",
)
subparser.add_argument(
"--seal_det_limit_side_len",
type=int,
help="This sets a limit on the side length of the input image for the seal text detection model.",
)
subparser.add_argument(
"--seal_det_limit_type",
type=str,
help="This determines how the side length limit is applied to the input image before feeding it into the seal text deteciton model.",
)
subparser.add_argument(
"--seal_det_thresh",
type=float,
help="Detection pixel threshold for the seal text detection model. Pixels with scores greater than this threshold in the output probability map are considered text pixels.",
)
subparser.add_argument(
"--seal_det_box_thresh",
type=float,
help="Detection box threshold for the seal 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.",
)
subparser.add_argument(
"--seal_det_unclip_ratio",
type=float,
help="Seal text detection expansion coefficient, which expands the text region using this method. The larger the value, the larger the expansion area.",
)
subparser.add_argument(
"--seal_text_recognition_model_name",
type=str,
help="Name of the seal text recognition model.",
)
subparser.add_argument(
"--seal_text_recognition_model_dir",
type=str,
help="Path to the seal text recognition model directory.",
)
subparser.add_argument(
"--seal_text_recognition_batch_size",
type=int,
help="Batch size for the seal text recognition model.",
)
subparser.add_argument(
"--seal_rec_score_thresh",
type=float,
help="Seal text recognition threshold. Text results with scores greater than this threshold are retained.",
)
subparser.add_argument(
"--formula_recognition_model_name",
type=str,
help="Name of the formula recognition model.",
)
subparser.add_argument(
"--formula_recognition_model_dir",
type=str,
help="Path to the formula recognition model directory.",
)
subparser.add_argument(
"--formula_recognition_batch_size",
type=int,
help="Batch size for the formula recognition model.",
)
subparser.add_argument(
"--use_doc_orientation_classify",
type=str2bool,
help="Whether to use document image orientation classification.",
)
subparser.add_argument(
"--use_doc_unwarping",
type=str2bool,
help="Whether to use text image unwarping.",
)
subparser.add_argument(
"--use_textline_orientation",
type=str2bool,
help="Whether to use text line orientation classification.",
)
subparser.add_argument(
"--use_seal_recognition",
type=str2bool,
help="Whether to use seal recognition.",
)
subparser.add_argument(
"--use_table_recognition",
type=str2bool,
help="Whether to use table recognition.",
)
subparser.add_argument(
"--use_formula_recognition",
type=str2bool,
help="Whether to use formula recognition.",
)
subparser.add_argument(
"--use_chart_recognition",
type=str2bool,
help="Whether to use chart recognition.",
)
subparser.add_argument(
"--use_region_detection",
type=str2bool,
help="Whether to use region detection.",
)
# FIXME: Passing API key through CLI is not secure; consider using
