admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-05-23 04:30:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(pix_fmt channels): use PyAv to check get pixel formats number of channels

Browse Source
This commit is contained in:
CarolinePascal
2026-05-22 02:01:05 +02:00
parent 72a429764a
commit 7498f1cf61
3 changed files with 31 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/datasets/test_depth.py
+4 -4
View File
@@ -25,7 +25,7 @@ from lerobot.datasets.image_writer import (
save_kwargs_for_path,
write_image,
)
from lerobot.datasets.video_utils import get_video_pixel_channels
from lerobot.datasets.pyav_utils import get_pix_fmt_channels
from tests.fixtures.constants import (
DEFAULT_FPS,
DUMMY_CAMERA_FEATURES,
@@ -195,9 +195,9 @@ class TestHwToDatasetFeaturesDepth:
class TestVideoInfoDepthFlag:
"""Misc depth-related constants and helpers in video_utils / configs."""
def test_get_video_pixel_channels_gray(self):
assert get_video_pixel_channels("gray12le") == 1
assert get_video_pixel_channels("gray8") == 1
def test_get_pix_fmt_channels_gray(self):
assert get_pix_fmt_channels("gray12le") == 1
assert get_pix_fmt_channels("gray8") == 1
def test_ffv1_in_valid_codecs(self):
assert "ffv1" in VALID_VIDEO_CODECS