admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-07-24 02:06:15 +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: else:
if arr.shape[-1] == 1: if arr.shape[-1] == 1:
img_entity = ( img_entity = (
rr.DepthImage(arr, colormap=rr.components.Colormap.Viridis).compress() rr.DepthImage(arr, colormap=rr.components.Colormap.Viridis)
if compress_images
else rr.DepthImage(arr, colormap=rr.components.Colormap.Viridis)
) )
else: else:
img_entity = rr.Image(arr).compress() if compress_images else rr.Image(arr) img_entity = rr.Image(arr).compress() if compress_images else rr.Image(arr)