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

fix(compress): removing rerun compression for DepthImages

Browse Source
This commit is contained in:
CarolinePascal
2026-06-24 17:46:43 +02:00
parent 73d15e160b
commit 31aa0766a2
1 changed files with 1 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lerobot/utils/visualization_utils.py
+1 -3
View File
@@ -109,9 +109,7 @@ def log_rerun_data(
else:
if arr.shape[-1] == 1:
img_entity = (
rr.DepthImage(arr, colormap=rr.components.Colormap.Viridis).compress()
if compress_images
else rr.DepthImage(arr, colormap=rr.components.Colormap.Viridis)
rr.DepthImage(arr, colormap=rr.components.Colormap.Viridis)
)
else:
img_entity = rr.Image(arr).compress() if compress_images else rr.Image(arr)