diff --git a/gluoncv/torch/data/video_cls/dataset_classification.py b/gluoncv/torch/data/video_cls/dataset_classification.py index f5f1da1729..b2f8473230 100644 --- a/gluoncv/torch/data/video_cls/dataset_classification.py +++ b/gluoncv/torch/data/video_cls/dataset_classification.py @@ -138,7 +138,7 @@ def __getitem__(self, index): sample = self.dataset_samples[index] buffer = self.loadvideo_decord(sample) buffer = self.data_transform(buffer) - return buffer, self.label_array[index], sample.split("/")[1].split(".")[0] + return buffer, self.label_array[index], sample.split("/")[-1].split(".")[0] elif self.mode == 'test': sample = self.test_dataset[index]