| |
| 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']) |
|
|
