mmtrack.datasets.dancetrack_dataset 源代码
# Copyright (c) OpenMMLab. All rights reserved.
from mmdet.datasets import DATASETS
from .mot_challenge_dataset import MOTChallengeDataset
[文档]@DATASETS.register_module()
class DanceTrackDataset(MOTChallengeDataset):
"""Dataset for DanceTrack: https://github.com/DanceTrack/DanceTrack.
Most content is inherited from MOTChallengeDataset.
"""
[文档] def get_benchmark_and_eval_split(self):
"""Get benchmark and dataset split to evaluate.
Get benchmark from upeper/lower-case image prefix and the dataset
split to evaluate.
Returns:
tuple(string): The first string denotes the type of dataset.
The second string denots the split of the dataset to eval.
"""
# As DanceTrack only has train/val and use 'val' for evaluation as
# default, we can directly output the desired split.
return 'DanceTrack', 'val'