mirror of
https://github.com/open-mmlab/mmocr.git
synced 2025-06-03 21:54:47 +08:00
6992923768
* [Enhance] Discard deprecated lmdb dataset format and only support img+label now * rename * update * add ut * updata document * update docs * update test * update test * Update dataset.md Co-authored-by: liukuikun <641417025@qq.com>
21 lines
746 B
Python
21 lines
746 B
Python
# Copyright (c) OpenMMLab. All rights reserved.
|
|
from unittest import TestCase
|
|
|
|
from mmocr.datasets import RecogLMDBDataset
|
|
|
|
|
|
class TestRecogLMDBDataset(TestCase):
|
|
|
|
def test_label_and_image_dataset(self):
|
|
|
|
# test initialization
|
|
dataset = RecogLMDBDataset(
|
|
ann_file='tests/data/rec_toy_dataset/imgs.lmdb', pipeline=[])
|
|
dataset.full_init()
|
|
self.assertEqual(len(dataset), 10)
|
|
self.assertEqual(len(dataset.load_data_list()), 10)
|
|
self.assertEqual(dataset[0]['img'].shape, (26, 67, 3))
|
|
self.assertEqual(dataset[0]['instances'][0]['text'], 'GRAND')
|
|
self.assertEqual(dataset[1]['img'].shape, (17, 37, 3))
|
|
self.assertEqual(dataset[1]['instances'][0]['text'], 'HOTEL')
|