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

feat(viz): render depth observations as rr.DepthImage in Viridis

Browse Source
This commit is contained in:
CarolinePascal
2026-05-20 16:22:34 +02:00
parent 14df709201
commit 1dafb4acf6
1 changed files with 8 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lerobot/utils/visualization_utils.py
+8 -1
View File
@@ -107,7 +107,14 @@ def log_rerun_data(
for i, vi in enumerate(arr): for i, vi in enumerate(arr):
rr.log(f"{key}_{i}", rr.Scalars(float(vi))) rr.log(f"{key}_{i}", rr.Scalars(float(vi)))
else: else:
img_entity = rr.Image(arr).compress() if compress_images else rr.Image(arr) if arr.shape[-1] == 1:
img_entity = (
rr.DepthImage(arr, colormap=rr.components.Colormap.Viridis).compress()
if compress_images
else rr.rr.DepthImage(arr, colormap=rr.components.Colormap.Viridis)
)
else:
img_entity = rr.Image(arr).compress() if compress_images else rr.Image(arr)
rr.log(key, entity=img_entity, static=True) rr.log(key, entity=img_entity, static=True)
if action: if action: