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

[feat] backend=av

Browse source
This commit is contained in:
zhengzangw 2024-05-30 08:51:22 +00:00
parent 6fe7aa36b9
commit ba6c800e9a
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
opensora/datasets/datasets.py
View file

@ -68,7 +68,7 @@ class VideoTextDataset(torch.utils.data.Dataset):
if file_type == "video":
# loading
vframes, vinfo = read_video(path, backend='cv2')
vframes, vinfo = read_video(path, backend='av')
video_fps = vinfo["video_fps"] if 'video_fps' in vinfo else 24
# Sampling video frames
@ -145,7 +145,7 @@ class VariableVideoTextDataset(VideoTextDataset):
video_fps = 24 # default fps
if file_type == "video":
# loading
vframes, vinfo = read_video(path, backend='cv2')
vframes, vinfo = read_video(path, backend='av')
video_fps = vinfo["video_fps"] if 'video_fps' in vinfo else 24
# Sampling video frames