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

VideoSimk automatically infers video_info

This commit is contained in:
LinasKo 2024-11-06 14:07:26 +02:00
parent 8bb609eca0
commit 659ddec360
1 changed files with 12 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
supervision/utils/video.py
View File

@ -65,8 +65,9 @@ class VideoSink:
Attributes:
target_path (str): The path to the output file where the video will be saved.
video_info (VideoInfo): Information about the video resolution, fps,
and total frame count.
video_info (Optional[VideoInfo]): Information about the output video resolution,
fps, and total frame count. If not provided, the information will be inferred
from the video path.
codec (str): FOURCC code for video format
Example:
@ -82,8 +83,16 @@ class VideoSink:
```
""" # noqa: E501 // docs
def __init__(self, target_path: str, video_info: VideoInfo, codec: str = "mp4v"):
def __init__(
self,
target_path: str,
video_info: Optional[VideoInfo] = None,
codec: str = "mp4v",
):
self.target_path = target_path
if video_info is None:
video_info = VideoInfo.from_video_path(target_path)
self.video_info = video_info
self.__codec = codec
self.__writer = None