Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2026-02-25 04:30:48 +01:00
Code Issues Packages Projects Releases Wiki Activity

fix

Browse Source
This commit is contained in:
xixing.tj
2022-09-02 15:22:20 +08:00
parent 13135b686c
commit 5cfaf122ef
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -150,7 +150,7 @@ class OCRDetectionPipeline(Pipeline):
rboxes = inputs['combined_rboxes'][0]
count = inputs['combined_counts'][0]
if count == 0 or count < rboxes.shape[0]:
raise Exception("No text detected")
raise Exception('No text detected')
rboxes = rboxes[:count, :]
# convert rboxes to polygons and find its coordinates on the original image