Spaces:
Sleeping
Sleeping
File size: 746 Bytes
0b4516f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
# 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')
|