From b8b08cacd3abbad4c6c883bb9151ec3a24bcaa97 Mon Sep 17 00:00:00 2001 From: SkalskiP Date: Tue, 7 Feb 2023 13:24:52 +0100 Subject: [PATCH] - make it simpler to import process_video function --- supervision/__init__.py | 2 +- supervision/video.py | 11 +++++------ 2 files changed, 6 insertions(+), 7 deletions(-) diff --git a/supervision/__init__.py b/supervision/__init__.py index bb6ee5b4..2d7228b4 100644 --- a/supervision/__init__.py +++ b/supervision/__init__.py @@ -8,4 +8,4 @@ from supervision.draw.utils import draw_filled_rectangle, draw_polygon, draw_tex from supervision.geometry.core import Point, Position, Rect from supervision.geometry.utils import get_polygon_center from supervision.notebook.utils import show_frame_in_notebook -from supervision.video import VideoInfo, VideoSink, get_video_frames_generator +from supervision.video import VideoInfo, VideoSink, get_video_frames_generator, process_video diff --git a/supervision/video.py b/supervision/video.py index 228b5a2d..970f96cb 100644 --- a/supervision/video.py +++ b/supervision/video.py @@ -20,7 +20,7 @@ class VideoInfo: Examples: ```python - >>> from supervision.video import VideoInfo + >>> from supervision import VideoInfo >>> video_info = VideoInfo.from_video_path(video_path='video.mp4') @@ -65,8 +65,7 @@ class VideoSink: Examples: ```python - >>> from supervision.video import VideoInfo - >>> from supervision.video import VideoSink + >>> from supervision import VideoInfo, VideoSink >>> video_info = VideoInfo.from_video_path(video_path='source_video.mp4') @@ -110,7 +109,7 @@ def get_video_frames_generator(source_path: str) -> Generator[np.ndarray, None, Examples: ```python - >>> from supervision.video import get_video_frames_generator + >>> from supervision import get_video_frames_generator >>> for frame in get_video_frames_generator(source_path='source_video.mp4'): ... ... @@ -141,9 +140,9 @@ def process_video( Examples: ```python - >>> from supervision.video import process_video + >>> from supervision import process_video - >>> def process_frame(frame: np.ndarray) -> np.ndarray: + >>> def process_frame(scene: np.ndarray) -> np.ndarray: ... ... >>> process_video(