mirrors
/
mmocr
mirror of https://github.com/open-mmlab/mmocr.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

[Feature] Extend ctw1500 converter to support text fields (#729) 

* Extend ctw1500 converter to support text fieldols/

* remove args for debug
pull/691/merge
liukuikun 2022-01-14 15:30:48 +08:00 committed by GitHub
parent 9de3348e9a
commit c736989615
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tools/data/textdet/ctw1500_converter.py
View File

@ -98,12 +98,14 @@ def load_txt_info(gt_file, img_info):
# convert to COCO style XYWH format # convert to COCO style XYWH format
min_x, min_y, max_x, max_y = polygon.bounds min_x, min_y, max_x, max_y = polygon.bounds
bbox = [min_x, min_y, max_x - min_x, max_y - min_y] bbox = [min_x, min_y, max_x - min_x, max_y - min_y]
text = strs[28][4:]
anno = dict( anno = dict(
iscrowd=iscrowd, iscrowd=iscrowd,
category_id=category_id, category_id=category_id,
bbox=bbox, bbox=bbox,
area=area, area=area,
text=text,
segmentation=[xy]) segmentation=[xy])
anno_info.append(anno) anno_info.append(anno)
img_info.update(anno_info=anno_info) img_info.update(anno_info=anno_info)
@ -120,7 +122,7 @@ def load_xml_info(gt_file, img_info):
w = box.attrib['width'] w = box.attrib['width']
x = box.attrib['left'] x = box.attrib['left']
y = box.attrib['top'] y = box.attrib['top']
# label = box[0].text text = box[0].text
segs = box[1].text segs = box[1].text
pts = segs.strip().split(',') pts = segs.strip().split(',')
pts = [int(x) for x in pts] pts = [int(x) for x in pts]
@ -141,6 +143,7 @@ def load_xml_info(gt_file, img_info):
category_id=category_id, category_id=category_id,
bbox=bbox, bbox=bbox,
area=area, area=area,
text=text,
segmentation=[pts]) segmentation=[pts])
anno_info.append(anno) anno_info.append(anno)