Spaces:
Sleeping
Sleeping
File size: 1,169 Bytes
9bf4bd7 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 |
# Copyright (c) OpenMMLab. All rights reserved.
import json
from typing import Tuple
from mmocr.registry import DATA_PARSERS
from mmocr.utils import bbox2poly
from .base import BaseParser
@DATA_PARSERS.register_module()
class FUNSDTextDetAnnParser(BaseParser):
"""FUNSD Text Detection Annotation Parser. See
dataset_zoo/funsd/sample_anno.md for annotation example.
Args:
nproc (int): The number of processes to parse the annotation. Defaults
to 1.
"""
def parse_file(self, img_path: str, ann_path: str) -> Tuple:
"""Parse single annotation."""
instances = list()
for poly, text, ignore in self.loader(ann_path):
instances.append(dict(poly=poly, text=text, ignore=ignore))
return img_path, instances
def loader(self, file_path: str):
with open(file_path, 'r') as f:
data = json.load(f)
for form in data['form']:
for word in form['words']:
poly = bbox2poly(word['box']).tolist()
text = word['text']
ignore = len(text) == 0
yield poly, text, ignore
|