Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2026-02-24 20:19:51 +01:00
Code Issues Packages Projects Releases Wiki Activity

update

Browse Source
This commit is contained in:
Yingda Chen
2024-10-31 18:20:09 +08:00
parent 456958ac1b
commit f8734b7214
1 changed files with 2 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
modelscope/pipelines/cv/ocr_detection_pipeline.py
View File

@@ -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(