nicolabs/Open-Sora
1
0
Fork 0
mirror of https://github.com/hpcaitech/Open-Sora.git synced 2026-05-21 11:59:01 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

[fix] dataset ext judge

Browse source
This commit is contained in:
Zangwei Zheng 2024-04-08 20:42:38 +08:00
parent 3210a691cf
commit 8d3f0f8282
1 changed files with 3 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
opensora/datasets/utils.py
View file

@ -1,3 +1,5 @@
import os
import numpy as np
import torch
import torchvision
@ -94,7 +96,7 @@ def read_video_from_path(path, transform=None, image_size=(256, 256)):
def read_from_path(path, image_size):
ext = path.split(".")[-1]
ext = os.path.splitext(path)[-1].lower()
if ext.lower() in VID_EXTENSIONS:
return read_video_from_path(path, image_size=image_size)
else: