diff --git a/tools/datasets/convert.py b/tools/datasets/convert.py index 6253e8e..fad128f 100644 --- a/tools/datasets/convert.py +++ b/tools/datasets/convert.py @@ -6,7 +6,7 @@ import pandas as pd from torchvision.datasets import ImageNet IMG_EXTENSIONS = (".jpg", ".jpeg", ".png", ".ppm", ".bmp", ".pgm", ".tif", ".tiff", ".webp") -VID_EXTENSIONS = (".mp4", ".avi", ".mov", ".mkv") +VID_EXTENSIONS = (".mp4", ".avi", ".mov", ".mkv", ".m2ts") def scan_recursively(root):