Spaces:
Runtime error
Runtime error
Commit
·
41f8ba0
1
Parent(s):
5f37adc
Update app.py
Browse files
app.py
CHANGED
|
@@ -20,7 +20,7 @@ def sample_uniform_frame_indices(clip_len, seg_len):
|
|
| 20 |
return np.array(indices).astype(np.int64)
|
| 21 |
|
| 22 |
def read_video_decord(file_path, indices):
|
| 23 |
-
vr = VideoReader(file_path, num_threads=1, ctx=gpu(0) if torch.cuda.is_available() else cpu(0))
|
| 24 |
video = vr.get_batch(indices).asnumpy()
|
| 25 |
return video
|
| 26 |
|
|
|
|
| 20 |
return np.array(indices).astype(np.int64)
|
| 21 |
|
| 22 |
def read_video_decord(file_path, indices):
|
| 23 |
+
vr = VideoReader(file_path, num_threads=1, ctx=cpu(0))#gpu(0) if torch.cuda.is_available() else cpu(0))
|
| 24 |
video = vr.get_batch(indices).asnumpy()
|
| 25 |
return video
|
| 26 |
|