def getData(self, index): """ Load ICDAR2019ArT data :param index: zero-based data index :return: A dict like { img: RGB, bboxes: nxkx2 np array, tags: n } """ img_path = self.img_paths[index] # RGB img = get_img(img_path) if self.split == 'test': return {'img': img, 'path': img_path} # get gt img_name = os.path.basename(img_path).split('.')[0] gt = self.gt[img_name] bboxes, tags, langs, trans = parse_gt(gt) item = { 'img': img, 'type': 'quad', 'bboxes': bboxes, 'lang': langs, 'trans': trans, 'tags': tags, 'path': img_path } return item
def getData(self, index): """ Load MSRA-TD500 data :param index: zero-based data index :return: A dict like { img: RGB, bboxes: nxkx2 np array, tags: n } """ img_path = self.img_paths[index] gt_path = self.gt_paths[index] # RGB img = get_img(img_path) # bbox normed to 0~1 bboxes, tags = get_bboxes(img, gt_path) item = {'img': img, 'type': 'contour', 'bboxes': bboxes, 'tags': tags, 'path': img_path} return item
def getData(self, index): """ Load ICDAR2019MLT data :param index: zero-based data index :return: A dict like { img: RGB, bboxes: nxkx2 np array, tags: n } """ img_path = self.img_paths[index] # RGB img = get_img(img_path) if self.split == 'test': return {'img': img, 'path': img_path} gt_path = self.gt_paths[index] # bbox normed to 0~1 bboxes, tags, langs, trans = get_bboxes(img, gt_path) # scale it back to pixel coord item = {'img': img, 'type': 'quad', 'bboxes': bboxes, 'lang': langs, 'trans': trans, 'tags': tags, 'path': img_path} return item
def getData(self, index): """ Load ICDAR2019ReCTS data :param index: zero-based data index :return: A dict like { img: RGB, bboxes: nxkx2 np array, tags: n } """ img_path = self.img_paths[index] # RGB img = get_img(img_path) if self.split == 'test': return {'img': img, 'path': img_path} # get gt gt_path = self.gt_paths[index] bboxes, tags, trans = parse_gt(gt_path) item = { 'img': img, 'type': 'quad', 'bboxes': bboxes, 'trans': trans, 'tags': tags, 'path': img_path } return item