File size: 1,916 Bytes
d3dbf03 |
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 |
# Copyright (c) OpenMMLab. All rights reserved.
from mmengine.testing import assert_dict_has_keys
from mmaction.datasets import VideoDataset
from mmaction.utils import register_all_modules
from .base import BaseTestDataset
class TestVideoDataset(BaseTestDataset):
register_all_modules()
def test_video_dataset(self):
video_dataset = VideoDataset(
self.video_ann_file,
self.video_pipeline,
data_prefix={'video': self.data_prefix},
start_index=3)
assert len(video_dataset) == 2
assert video_dataset.start_index == 3
video_dataset = VideoDataset(
self.video_ann_file,
self.video_pipeline,
data_prefix={'video': self.data_prefix})
assert video_dataset.start_index == 0
def test_video_dataset_multi_label(self):
video_dataset = VideoDataset(
self.video_ann_file_multi_label,
self.video_pipeline,
data_prefix={'video': self.data_prefix},
multi_class=True,
num_classes=100)
assert video_dataset.start_index == 0
def test_video_pipeline(self):
target_keys = ['filename', 'label', 'start_index', 'modality']
# VideoDataset not in test mode
video_dataset = VideoDataset(
self.video_ann_file,
self.video_pipeline,
data_prefix={'video': self.data_prefix},
test_mode=False)
result = video_dataset[0]
assert assert_dict_has_keys(result, target_keys)
# VideoDataset in test mode
video_dataset = VideoDataset(
self.video_ann_file,
self.video_pipeline,
data_prefix={'video': self.data_prefix},
test_mode=True)
result = video_dataset[0]
assert assert_dict_has_keys(result, target_keys)
|