admin/any4lerobot
1
0
Fork 0
mirror of https://github.com/Tavish9/any4lerobot.git synced 2026-05-23 17:59:41 +00:00
Code Issues Packages Projects Releases Wiki Activity

add preprocess method for bc_z dataset (#2)

Browse Source
This commit is contained in:
Chopin
2025-02-23 10:28:30 +08:00
committed by GitHub
parent 8a720b6ae6
commit 69660941d0
1 changed files with 12 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
README.md
+12 -1
View File
@@ -83,7 +83,7 @@ pip install -e .
## Get started ## Get started
> [!IMPORTANT] > [!IMPORTANT]
> Before running the following code, modify `consolidate()` function in lerobot. > 1.Before running the following code, modify `consolidate()` function in lerobot.
> ```python > ```python
> def consolidate(self, run_compute_stats: bool = True, keep_image_files: bool = False, stat_kwargs: dict = {}) -> None: > def consolidate(self, run_compute_stats: bool = True, keep_image_files: bool = False, stat_kwargs: dict = {}) -> None:
> ... > ...
@@ -93,6 +93,17 @@ pip install -e .
> self.meta.stats = compute_stats(self, **stat_kwargs) > self.meta.stats = compute_stats(self, **stat_kwargs)
> ... > ...
> ``` > ```
> 2.for `bc_z` dataset, two source codes need to be modified.
>
> path: `lerobot/common/datasets/video_utils.py`
>
> method: `encode_video_frames`
> ```python
> # add the following content to line 141:
> vf: str = "pad=ceil(iw/2)*2:ceil(ih/2)*2",
> # Add the following content to line 171:
> ffmpeg_args["-vf"] = vf
> ```
Download source code: Download source code: