xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2026-05-19 11:27:03 +08:00
Code Issues Packages Projects Releases Wiki Activity

[bugfix] Lazy import cv2 (#26869)

Browse Source 
Signed-off-by: angelayi <yiangela7@gmail.com>
This commit is contained in:
Angela Yi 2025-10-15 00:47:50 -07:00 committed by GitHub
parent b8a4572157
commit efdef57b1f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
vllm/assets/video.py
View File

@ -5,7 +5,6 @@ from dataclasses import dataclass
from functools import lru_cache from functools import lru_cache
from typing import Any, ClassVar, Literal from typing import Any, ClassVar, Literal
import cv2
import numpy as np import numpy as np
import numpy.typing as npt import numpy.typing as npt
from huggingface_hub import hf_hub_download from huggingface_hub import hf_hub_download
@ -43,6 +42,8 @@ def download_video_asset(filename: str) -> str:
def video_to_ndarrays(path: str, num_frames: int = -1) -> npt.NDArray: def video_to_ndarrays(path: str, num_frames: int = -1) -> npt.NDArray:
import cv2
cap = cv2.VideoCapture(path) cap = cv2.VideoCapture(path)
if not cap.isOpened(): if not cap.isOpened():
raise ValueError(f"Could not open video file {path}") raise ValueError(f"Could not open video file {path}")
@ -78,6 +79,8 @@ def video_to_pil_images_list(path: str, num_frames: int = -1) -> list[Image.Imag
def video_get_metadata(path: str, num_frames: int = -1) -> dict[str, Any]: def video_get_metadata(path: str, num_frames: int = -1) -> dict[str, Any]:
import cv2
cap = cv2.VideoCapture(path) cap = cv2.VideoCapture(path)
if not cap.isOpened(): if not cap.isOpened():
raise ValueError(f"Could not open video file {path}") raise ValueError(f"Could not open video file {path}")