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