File size: 1,665 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 |
# Copyright (c) OpenMMLab. All rights reserved.
from mmengine.testing import assert_dict_has_keys
from mmaction.datasets import VideoTextDataset
from mmaction.utils import register_all_modules
from .base import BaseTestDataset
class TestVideoTextDataset(BaseTestDataset):
register_all_modules()
def test_video_dataset(self):
video_dataset = VideoTextDataset(
self.video_text_ann_file,
self.video_text_pipeline,
data_prefix={'video': self.data_prefix},
start_index=3)
assert len(video_dataset) == 2
assert video_dataset.start_index == 3
video_dataset = VideoTextDataset(
self.video_text_ann_file,
self.video_text_pipeline,
data_prefix={'video': self.data_prefix})
assert video_dataset.start_index == 0
def test_video_pipeline(self):
target_keys = ['filename', 'text', 'start_index', 'modality', 'imgs']
# VideoTextDataset not in test mode
video_dataset = VideoTextDataset(
self.video_text_ann_file,
self.video_text_pipeline,
data_prefix={'video': self.data_prefix},
test_mode=False)
result = video_dataset[0]
assert assert_dict_has_keys(result, target_keys)
# VideoTextDataset in test mode
video_dataset = VideoTextDataset(
self.video_text_ann_file,
self.video_text_pipeline,
data_prefix={'video': self.data_prefix},
test_mode=True)
result = video_dataset[0]
assert assert_dict_has_keys(result, target_keys)
|