Unverified Commit efdef57b authored by Angela Yi's avatar Angela Yi Committed by GitHub
Browse files

[bugfix] Lazy import cv2 (#26869)


Signed-off-by: default avatarangelayi <yiangela7@gmail.com>
parent b8a45721
...@@ -5,7 +5,6 @@ from dataclasses import dataclass ...@@ -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: ...@@ -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 ...@@ -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}")
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment