remove hardcoded video extension in kinetics400 dataset (#1418)

929c81d4 · Zhicheng Yan · Francisco Massa · 76702a03 · 929c81d4
Commit 929c81d4 authored Oct 06, 2019 by Zhicheng Yan Committed by Francisco Massa Oct 07, 2019
Hide whitespace changes
Inline Side-by-side

Showing with 0 additions and 1 deletion

torchvision/datasets/kinetics.py torchvision/datasets/kinetics.py +0 -1

No files found.
--- a/torchvision/datasets/kinetics.py
+++ b/torchvision/datasets/kinetics.py
@@ -41,7 +41,6 @@ class Kinetics400(VisionDataset):
                 num_workers=1, _video_width=0, _video_height=0,
                 _video_min_dimension=0, _audio_samples=0):
        super(Kinetics400, self).__init__(root)
-        extensions = ('avi',)
        classes = list(sorted(list_dir(root)))
        class_to_idx = {classes[i]: i for i in range(len(classes))}