機械学習の人に人気なLeRobot So101を作ってみる　その６：データセットの登録

トークンを登録

Hugging Face – The AI community building the future.

HuggingFaceCLIで登録

これまでの手順を実施していれば入っている筈ですが、無い場合は

$ pip install -U "huggingface_hub[cli]"

され、ここまでやったあと

$ HF_TOKEN=YOURKEY
$ huggingface-cli login --token ${HF_TOKEN} --add-to-git-credential

環境変数を導入

$ HF_USER=$(huggingface-cli whoami | head -n 1)
$ echo $HF_USER

この値を~/.bashrcなどに

$ export HF_USER=foobar

以下のコマンドでデータ登録とアップロードをする素敵仕様
記録時間は６０秒（デフォ値と同じ）、dataset.num_episodes変数が回数で、今回は２回

$ python3 -m lerobot.record \
    --robot.type=so101_follower \
    --robot.port=/dev/ttyACM1 \
    --robot.id=my_awesome_follower_arm \
    --robot.cameras="{ front: {type: opencv, index_or_path: 0, width: 1920, height: 1080, fps: 30}}" \
    --teleop.type=so101_leader \
    --teleop.port=/dev/ttyACM2 \
    --teleop.id=my_awesome_leader_arm \
    --display_data=true \
    --dataset.repo_id=${HF_USER}/record-test \
    --dataset.num_episodes=2 \
    --dataset.episode_time_s=60 \
    --dataset.single_task="Grab the rubber duck"

データセットは"~/.cache/huggingface/lerobot/{repo-id}"に上がってる

練習したいときは"--dataset.push_to_hub=False"のオプションを追加