admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-05-27 06:29:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(video): add ffv1 to supported codecs

Browse Source
This commit is contained in:
CarolinePascal
2026-05-19 15:48:27 +02:00
parent 42214d1c7a
commit 088352383d
1 changed files with 7 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lerobot/configs/video.py
+7 -1
View File
@@ -39,7 +39,9 @@ HW_VIDEO_CODECS = [
"h264_vaapi", # Linux Intel/AMD "h264_vaapi", # Linux Intel/AMD
"h264_qsv", # Intel Quick Sync "h264_qsv", # Intel Quick Sync
] ]
VALID_VIDEO_CODECS: frozenset[str] = frozenset({"h264", "hevc", "libsvtav1", "auto", *HW_VIDEO_CODECS}) VALID_VIDEO_CODECS: frozenset[str] = frozenset(
{"h264", "hevc", "libsvtav1", "ffv1", "auto", *HW_VIDEO_CODECS}
)
# Aliases for legacy video codec names. # Aliases for legacy video codec names.
VIDEO_CODECS_ALIASES: dict[str, str] = {"av1": "libsvtav1"} VIDEO_CODECS_ALIASES: dict[str, str] = {"av1": "libsvtav1"}
@@ -229,6 +231,10 @@ class VideoEncoderConfig:
elif self.vcodec == "h264_qsv": elif self.vcodec == "h264_qsv":
set_if("global_quality", self.crf) set_if("global_quality", self.crf)
set_if("preset", self.preset) set_if("preset", self.preset)
elif self.vcodec == "ffv1":
# Lossless intra-frame codec. ``crf``/``preset``/``fast_decode``
# are not meaningful.
set_if("threads", encoder_threads)
else: else:
set_if("crf", self.crf) set_if("crf", self.crf)
set_if("preset", self.preset) set_if("preset", self.preset)