Skip to content

Commit 929c81d

Browse files
stephenyan1231fmassa
authored andcommitted
remove hardcoded video extension in kinetics400 dataset (#1418)
1 parent 76702a0 commit 929c81d

File tree

1 file changed

+0
-1
lines changed

1 file changed

+0
-1
lines changed

torchvision/datasets/kinetics.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -41,7 +41,6 @@ def __init__(self, root, frames_per_clip, step_between_clips=1, frame_rate=None,
4141
num_workers=1, _video_width=0, _video_height=0,
4242
_video_min_dimension=0, _audio_samples=0):
4343
super(Kinetics400, self).__init__(root)
44-
extensions = ('avi',)
4544

4645
classes = list(sorted(list_dir(root)))
4746
class_to_idx = {classes[i]: i for i in range(len(classes))}

0 commit comments

Comments
 (0)