From 310ce7d23d03d4fb256185c635ec13dcc5df934f Mon Sep 17 00:00:00 2001 From: Yingda Chen Date: Wed, 27 Nov 2024 22:45:39 +0800 Subject: [PATCH] fix typo --- modelscope/pipelines/cv/ocr_detection_pipeline.py | 14 ++++---------- 1 file changed, 4 insertions(+), 10 deletions(-) diff --git a/modelscope/pipelines/cv/ocr_detection_pipeline.py b/modelscope/pipelines/cv/ocr_detection_pipeline.py index 93420a28..5b0fbda5 100644 --- a/modelscope/pipelines/cv/ocr_detection_pipeline.py +++ b/modelscope/pipelines/cv/ocr_detection_pipeline.py @@ -89,16 +89,10 @@ class OCRDetectionPipeline(Pipeline): tf = tf.compat.v1 tf.compat.v1.disable_eager_execution() - tf.app.flags.DEFINE_float( - 'node_threshold', - TF_NODE_THRESHOLD, - 'Confidence threshold for nodes', - already_declared_error=False) - tf.app.flags.DEFINE_float( - 'link_threshold', - TF_LINK_THRESHOLD, - 'Confidence threshold for links', - already_declared_error=False) + tf.app.flags.DEFINE_float('node_threshold', TF_NODE_THRESHOLD, + 'Confidence threshold for nodes') + tf.app.flags.DEFINE_float('link_threshold', TF_LINK_THRESHOLD, + 'Confidence threshold for links') tf.reset_default_graph() model_path = osp.join( osp.join(self.model, ModelFile.TF_CHECKPOINT_FOLDER),