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

tests(depth): adding new tests for depth integration validation

Browse Source
This commit is contained in:
CarolinePascal
2026-05-21 20:20:40 +02:00
parent 4ea8653ca3
commit 72a429764a
4 changed files with 343 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/fixtures/constants.py
Vendored
+11
View File
@@ -46,5 +46,16 @@ DUMMY_CAMERA_FEATURES = {
"laptop": {"shape": (64, 96, 3), "names": ["height", "width", "channels"], "info": DUMMY_VIDEO_INFO},
"phone": {"shape": (64, 96, 3), "names": ["height", "width", "channels"], "info": DUMMY_VIDEO_INFO},
}
DUMMY_DEPTH_VIDEO_INFO = {
**DUMMY_VIDEO_INFO,
"is_depth_map": True,
}
DUMMY_DEPTH_CAMERA_FEATURES = {
"laptop_depth": {
"shape": (64, 96, 1),
"names": ["height", "width", "channels"],
"info": DUMMY_DEPTH_VIDEO_INFO,
},
}
DUMMY_CHW = (3, 96, 128)
DUMMY_HWC = (96, 128, 3)