Compare commits
4 Commits
aa0bcaffea
...
2f9bc549a4
Author | SHA1 | Date |
---|---|---|
|
2f9bc549a4 | |
|
b568de35ad | |
|
ae9c81ac39 | |
|
6d0e058186 |
|
@ -98,7 +98,7 @@ conda create -y -n lerobot python=3.10
|
||||||
conda activate lerobot
|
conda activate lerobot
|
||||||
```
|
```
|
||||||
|
|
||||||
When using `miniconda`, if you don't have `fffmpeg` in your environment:
|
When using `miniconda`, if you don't have `ffmpeg` in your environment:
|
||||||
```bash
|
```bash
|
||||||
conda install ffmpeg
|
conda install ffmpeg
|
||||||
```
|
```
|
||||||
|
|
|
@ -27,7 +27,6 @@ import torch
|
||||||
import torch.utils
|
import torch.utils
|
||||||
from datasets import concatenate_datasets, load_dataset
|
from datasets import concatenate_datasets, load_dataset
|
||||||
from huggingface_hub import HfApi, snapshot_download
|
from huggingface_hub import HfApi, snapshot_download
|
||||||
from huggingface_hub.constants import REPOCARD_NAME
|
|
||||||
from huggingface_hub.errors import RevisionNotFoundError
|
from huggingface_hub.errors import RevisionNotFoundError
|
||||||
|
|
||||||
from lerobot.common.constants import HF_LEROBOT_HOME
|
from lerobot.common.constants import HF_LEROBOT_HOME
|
||||||
|
@ -560,11 +559,10 @@ class LeRobotDataset(torch.utils.data.Dataset):
|
||||||
else:
|
else:
|
||||||
hub_api.upload_folder(**upload_kwargs)
|
hub_api.upload_folder(**upload_kwargs)
|
||||||
|
|
||||||
if not hub_api.file_exists(self.repo_id, REPOCARD_NAME, repo_type="dataset", revision=branch):
|
card = create_lerobot_dataset_card(
|
||||||
card = create_lerobot_dataset_card(
|
tags=tags, dataset_info=self.meta.info, license=license, **card_kwargs
|
||||||
tags=tags, dataset_info=self.meta.info, license=license, **card_kwargs
|
)
|
||||||
)
|
card.push_to_hub(repo_id=self.repo_id, repo_type="dataset", revision=branch)
|
||||||
card.push_to_hub(repo_id=self.repo_id, repo_type="dataset", revision=branch)
|
|
||||||
|
|
||||||
if tag_version:
|
if tag_version:
|
||||||
with contextlib.suppress(RevisionNotFoundError):
|
with contextlib.suppress(RevisionNotFoundError):
|
||||||
|
|
|
@ -257,6 +257,7 @@ def encode_video_frames(
|
||||||
) -> None:
|
) -> None:
|
||||||
"""More info on ffmpeg arguments tuning on `benchmark/video/README.md`"""
|
"""More info on ffmpeg arguments tuning on `benchmark/video/README.md`"""
|
||||||
video_path = Path(video_path)
|
video_path = Path(video_path)
|
||||||
|
imgs_dir = Path(imgs_dir)
|
||||||
video_path.parent.mkdir(parents=True, exist_ok=True)
|
video_path.parent.mkdir(parents=True, exist_ok=True)
|
||||||
|
|
||||||
ffmpeg_args = OrderedDict(
|
ffmpeg_args = OrderedDict(
|
||||||
|
|
Loading…
Reference in New Issue