Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2025-12-25 12:39:25 +01:00
Code Issues Packages Projects Releases Wiki Activity

fix potential double definition for ocr pipeline (#1102)

Browse Source 
* fix potential double definition issue
---------

Co-authored-by: Yingda Chen <yingda.chen@alibaba-inc.com>
This commit is contained in:
Yingda Chen
2024-11-28 00:43:24 +08:00
committed by GitHub
parent a7856a5995
commit e2bd302175
1 changed files with 5 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -27,6 +27,8 @@ RBOX_DIM = 5
OFFSET_DIM = 6
WORD_POLYGON_DIM = 8
OFFSET_VARIANCE = [0.1, 0.1, 0.1, 0.1, 0.1, 0.1]
TF_NODE_THRESHOLD = 0.4
TF_LINK_THRESHOLD = 0.6
@PIPELINES.register_module(
@@ -39,7 +41,7 @@ class OCRDetectionPipeline(Pipeline):
```python
>>> from modelscope.pipelines import pipeline
>>> ocr_detection = pipeline('ocr_detection', model='damo/cv_resnet18_ocr-detection-line-level_damo')
>>> ocr_detection = pipeline('ocr-detection', model='damo/cv_resnet18_ocr-detection-line-level_damo')
>>> result = ocr_detection('https://modelscope.oss-cn-beijing.aliyuncs.com/test/images/ocr_detection.jpg')
{'polygons': array([[220, 14, 780, 14, 780, 64, 220, 64],
@@ -87,9 +89,9 @@ class OCRDetectionPipeline(Pipeline):
tf = tf.compat.v1
tf.compat.v1.disable_eager_execution()
tf.app.flags.DEFINE_float('node_threshold', 0.4,
tf.app.flags.DEFINE_float('node_threshold', TF_NODE_THRESHOLD,
'Confidence threshold for nodes')
tf.app.flags.DEFINE_float('link_threshold', 0.6,
tf.app.flags.DEFINE_float('link_threshold', TF_LINK_THRESHOLD,
'Confidence threshold for links')
tf.reset_default_graph()
model_path = osp.join(
@@ -192,18 +194,6 @@ class OCRDetectionPipeline(Pipeline):
return result
else:
# for model seglink++
import tensorflow as tf
if tf.__version__ >= '2.0':
tf = tf.compat.v1
tf.compat.v1.disable_eager_execution()
tf.app.flags.DEFINE_float('node_threshold', 0.4,
'Confidence threshold for nodes')
tf.app.flags.DEFINE_float('link_threshold', 0.6,
'Confidence threshold for links')
img = LoadImage.convert_to_ndarray(input)
h, w, c = img.shape