xiaoxue_admin
/
supervision
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

- make it simpler to import process_video function

This commit is contained in:
SkalskiP 2023-02-07 13:24:52 +01:00
parent c8868c3412
commit b8b08cacd3
2 changed files with 6 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
supervision/__init__.py
View File

@ -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

11
supervision/video.py
View File

@ -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(