File size: 1,439 Bytes
6c9ac8f |
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 unittest
from mmdet.datasets import OpenImagesChallengeDataset, OpenImagesDataset
class TestOpenImagesDataset(unittest.TestCase):
def test_init(self):
dataset = OpenImagesDataset(
data_root='tests/data/OpenImages/',
ann_file='annotations/oidv6-train-annotations-bbox.csv',
data_prefix=dict(img='OpenImages/train/'),
label_file='annotations/class-descriptions-boxable.csv',
hierarchy_file='annotations/bbox_labels_600_hierarchy.json',
meta_file='annotations/image-metas.pkl',
pipeline=[])
dataset.full_init()
self.assertEqual(len(dataset), 1)
self.assertEqual(dataset.metainfo['classes'], ['Airplane'])
class TestOpenImagesChallengeDataset(unittest.TestCase):
def test_init(self):
dataset = OpenImagesChallengeDataset(
data_root='tests/data/OpenImages/',
ann_file='challenge2019/challenge-2019-train-detection-bbox.txt',
data_prefix=dict(img='OpenImages/train/'),
label_file='challenge2019/cls-label-description.csv',
hierarchy_file='challenge2019/class_label_tree.np',
meta_file='annotations/image-metas.pkl',
pipeline=[])
dataset.full_init()
self.assertEqual(len(dataset), 1)
self.assertEqual(dataset.metainfo['classes'], ['Airplane'])
|