From f8734b7214ff187ead7420d0b312cb03c0283ca8 Mon Sep 17 00:00:00 2001 From: Yingda Chen Date: Thu, 31 Oct 2024 18:20:09 +0800 Subject: [PATCH] update --- modelscope/pipelines/cv/ocr_detection_pipeline.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/modelscope/pipelines/cv/ocr_detection_pipeline.py b/modelscope/pipelines/cv/ocr_detection_pipeline.py index 4e1ee4aa..c23f6e6e 100644 --- a/modelscope/pipelines/cv/ocr_detection_pipeline.py +++ b/modelscope/pipelines/cv/ocr_detection_pipeline.py @@ -18,9 +18,7 @@ from modelscope.utils.config import Config from modelscope.utils.constant import ModelFile, Tasks from modelscope.utils.device import device_placement from modelscope.utils.logger import get_logger -from .ocr_utils import (cal_width, combine_segments_python, - decode_segments_links_python, nms_python, - rboxes_to_polygons) +from .ocr_utils import cal_width, nms_python, rboxes_to_polygons logger = get_logger() @@ -119,7 +117,7 @@ class OCRDetectionPipeline(Pipeline): variable_averages = tf.train.ExponentialMovingAverage( 0.997, global_step) - from .ocr_utils import SegLinkDetector + from .ocr_utils import SegLinkDetector, combine_segments_python, decode_segments_links_python # detector detector = SegLinkDetector() all_maps = detector.build_model(