We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 06b914e commit 4728256Copy full SHA for 4728256
tensorrt_llm/inputs/utils.py
@@ -1,6 +1,5 @@
1
from typing import List, Union
2
3
-import cv2
4
import numpy as np
5
import requests
6
import torch
@@ -30,6 +29,8 @@ def load_video(
30
29
num_frames: int = 10,
31
format: str = "pt",
32
device: str = "cuda") -> Union[List[Image.Image], List[torch.Tensor]]:
+ import cv2
33
+
34
assert format in ["pt", "pil"], "format must be either Pytorch or PIL"
35
36
# Load video frames from a video file
0 commit comments