File size: 4,061 Bytes
d670799 |
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 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 |
# Copyright (c) OpenMMLab. All rights reserved.
from mmengine.testing import assert_dict_has_keys
from mmaction.datasets import RawframeDataset
from mmaction.utils import register_all_modules
from .base import BaseTestDataset
class TestRawframDataset(BaseTestDataset):
def test_rawframe_dataset(self):
rawframe_dataset = RawframeDataset(self.frame_ann_file,
self.frame_pipeline,
{'img': self.data_prefix})
assert rawframe_dataset.start_index == 1
def test_rawframe_dataset_with_offset(self):
register_all_modules()
rawframe_dataset = RawframeDataset(
self.frame_ann_file_with_offset,
self.frame_pipeline, {'img': self.data_prefix},
with_offset=True)
assert rawframe_dataset.start_index == 1
def test_rawframe_dataset_multi_label(self):
register_all_modules()
rawframe_dataset = RawframeDataset(
self.frame_ann_file_multi_label,
self.frame_pipeline, {'img': self.data_prefix},
multi_class=True,
num_classes=100)
assert rawframe_dataset.start_index == 1
def test_dataset_realpath(self):
register_all_modules()
dataset = RawframeDataset(self.frame_ann_file, self.frame_pipeline,
{'img': '.'})
dataset = RawframeDataset(self.frame_ann_file, self.frame_pipeline,
{'img': 's3://good'})
assert dataset.data_prefix == {'img': 's3://good'}
dataset = RawframeDataset(self.frame_ann_file, self.frame_pipeline)
def test_rawframe_pipeline(self):
target_keys = [
'frame_dir', 'total_frames', 'label', 'filename_tmpl',
'start_index', 'modality'
]
# RawframeDataset not in test mode
rawframe_dataset = RawframeDataset(
self.frame_ann_file,
self.frame_pipeline, {'img': self.data_prefix},
test_mode=False)
result = rawframe_dataset[0]
assert assert_dict_has_keys(result, target_keys)
# RawframeDataset in multi-class tasks
rawframe_dataset = RawframeDataset(
self.frame_ann_file,
self.frame_pipeline, {'img': self.data_prefix},
multi_class=True,
num_classes=400,
test_mode=False)
result = rawframe_dataset[0]
assert assert_dict_has_keys(result, target_keys)
# RawframeDataset with offset
rawframe_dataset = RawframeDataset(
self.frame_ann_file_with_offset,
self.frame_pipeline, {'img': self.data_prefix},
with_offset=True,
num_classes=400,
test_mode=False)
result = rawframe_dataset[0]
assert assert_dict_has_keys(result, target_keys + ['offset'])
# RawframeDataset in test mode
rawframe_dataset = RawframeDataset(
self.frame_ann_file,
self.frame_pipeline, {'img': self.data_prefix},
test_mode=True)
result = rawframe_dataset[0]
assert assert_dict_has_keys(result, target_keys)
# RawframeDataset in multi-class tasks in test mode
rawframe_dataset = RawframeDataset(
self.frame_ann_file,
self.frame_pipeline, {'img': self.data_prefix},
multi_class=True,
num_classes=400,
test_mode=True)
result = rawframe_dataset[0]
assert assert_dict_has_keys(result, target_keys)
# RawframeDataset with offset
rawframe_dataset = RawframeDataset(
self.frame_ann_file_with_offset,
self.frame_pipeline, {'img': self.data_prefix},
with_offset=True,
num_classes=400,
test_mode=True)
result = rawframe_dataset[0]
assert assert_dict_has_keys(result, target_keys + ['offset'])
|