xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2026-03-16 13:37:23 +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 typing import Any, ClassVar, Literal
import cv2
import numpy as np
import numpy.typing as npt
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:
import cv2
cap = cv2.VideoCapture(path)
if not cap.isOpened():
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]:
import cv2
cap = cv2.VideoCapture(path)
if not cap.isOpened():
raise ValueError(f"Could not open video file {path}")