nicolabs/Open-Sora
1
0
Fork 0
mirror of https://github.com/hpcaitech/Open-Sora.git synced 2026-04-17 14:25:07 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

added .m2ts support

Browse source
This commit is contained in:
Tom Young 2024-06-30 09:35:59 +00:00
parent 7ec46c4732
commit 08c5222cb9
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
tools/datasets/convert.py
View file

@ -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):