wangzixiang
/
lerobot
mirror of https://github.com/huggingface/lerobot.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

update script names

This commit is contained in:
Thomas Wolf 2024-05-30 23:24:27 +02:00
parent 265b0ec44d
commit 0f50e033e9
1 changed files with 10 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
lerobot/scripts/push_dataset_to_hub.py
View File

@ -144,7 +144,8 @@ def push_videos_to_hub(repo_id, videos_dir, revision):
def push_dataset_to_hub( def push_dataset_to_hub(
data_dir: Path, raw_dir: Path,
out_dir: Path,
dataset_id: str, dataset_id: str,
raw_format: str | None, raw_format: str | None,
community_id: str, community_id: str,
@ -161,9 +162,6 @@ def push_dataset_to_hub(
): ):
repo_id = f"{community_id}/{dataset_id}" repo_id = f"{community_id}/{dataset_id}"
raw_dir = data_dir / f"{dataset_id}_raw"
out_dir = data_dir / repo_id
meta_data_dir = out_dir / "meta_data" meta_data_dir = out_dir / "meta_data"
videos_dir = out_dir / "videos" videos_dir = out_dir / "videos"
@ -245,10 +243,16 @@ def main():
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser()
parser.add_argument( parser.add_argument(
"--data-dir", "--raw-dir",
type=Path, type=Path,
required=True, required=True,
help="Root directory containing datasets (e.g. `data` or `tmp/data` or `/tmp/lerobot/data`).", help="Directory containing input raw datasets (e.g. `data/aloha_mobile_chair_raw` or `data/pusht_raw).",
)
parser.add_argument(
"--out-dir",
type=Path,
required=True,
help="Root directory containing output dataset (e.g. `data/lerobot/aloha_mobile_chair` or `data/lerobot/pusht`).",
) )
parser.add_argument( parser.add_argument(
"--dataset-id", "--dataset-id",