Commit
•
b826922
1
Parent(s):
9ea02f2
fix video processing bug
Browse files
data.py
CHANGED
|
@@ -327,6 +327,7 @@ def load_and_transform_video_data(
|
|
| 327 |
decode_audio=False,
|
| 328 |
**{"sample_rate": sample_rate},
|
| 329 |
)
|
|
|
|
| 330 |
|
| 331 |
all_clips_timepoints = get_clip_timepoints(clip_sampler, video.duration)
|
| 332 |
|
|
|
|
| 327 |
decode_audio=False,
|
| 328 |
**{"sample_rate": sample_rate},
|
| 329 |
)
|
| 330 |
+
video = torch.from_numpy(video)
|
| 331 |
|
| 332 |
all_clips_timepoints = get_clip_timepoints(clip_sampler, video.duration)
|
| 333 |
|