mirrors/mmocr
1
0
Fork 0
mirror of https://github.com/open-mmlab/mmocr.git synced 2025-06-03 21:54:47 +08:00
Code Issues Projects Releases Wiki Activity

[Fix] bezier_to_polygon -> bezier2polygon (#1739)

Browse Source
This commit is contained in:
double22a 2023-02-21 19:16:17 +08:00 committed by GitHub
parent 6eaa0673f7
commit 280a89c18e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tools/dataset_converters/common/curvedsyntext_converter.py
View File

@ -6,7 +6,7 @@ from functools import partial
import mmengine import mmengine
import numpy as np import numpy as np
from mmocr.utils import bezier_to_polygon, sort_points from mmocr.utils import bezier2polygon, sort_points
# The default dictionary used by CurvedSynthText # The default dictionary used by CurvedSynthText
dict95 = [ dict95 = [
@ -38,7 +38,7 @@ def digit2text(rec):
def modify_annotation(ann, num_sample, start_img_id=0, start_ann_id=0): def modify_annotation(ann, num_sample, start_img_id=0, start_ann_id=0):
ann['text'] = digit2text(ann.pop('rec')) ann['text'] = digit2text(ann.pop('rec'))
# Get hide egmentation points # Get hide egmentation points
polygon_pts = bezier_to_polygon(ann['bezier_pts'], num_sample=num_sample) polygon_pts = bezier2polygon(ann['bezier_pts'], num_sample=num_sample)
ann['segmentation'] = np.asarray(sort_points(polygon_pts)).reshape( ann['segmentation'] = np.asarray(sort_points(polygon_pts)).reshape(
1, -1).tolist() 1, -1).tolist()
ann['image_id'] += start_img_id ann['image_id'] += start_img_id