admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-05-22 20:19:43 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add resume

Browse Source
This commit is contained in:
glannuzel
2025-08-07 11:29:15 +02:00
parent 33b6a4ddcd
commit 20f5f45ac2
1 changed files with 21 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lerobot/robots/reachy2/data_acquisition_server.py
+21 -11
View File
@@ -211,18 +211,28 @@ class DataAcquisitionServicer():
obs_features = hw_to_dataset_features(self.robot.observation_features, "observation") obs_features = hw_to_dataset_features(self.robot.observation_features, "observation")
dataset_features = {**action_features, **obs_features} dataset_features = {**action_features, **obs_features}
# Create the dataset print(f"Dataset name: {request.dataset_name}")
self.dataset = LeRobotDataset.create(
repo_id=request.dataset_name,
fps=self.fps,
features=dataset_features,
robot_type=self.robot.name,
use_videos=True,
image_writer_threads=4,
)
current_dataset = self.create_dataset(request.dataset_name, DatasetPushState.LOCAL_ONLY, 0) if request.resume:
self.add_dataset_to_json_file(current_dataset, self.dataset_list_path) self.dataset = LeRobotDataset(
request.dataset_name,
root="/home/demo/.cache/huggingface/lerobot/" + request.dataset_name,
)
if hasattr(self.robot, "cameras") and len(self.robot.cameras) > 0:
self.dataset.start_image_writer()
else:
# Create the dataset
self.dataset = LeRobotDataset.create(
repo_id=request.dataset_name,
fps=self.fps,
features=dataset_features,
robot_type=self.robot.name,
use_videos=True,
image_writer_threads=4,
)
current_dataset = self.create_dataset(request.dataset_name, DatasetPushState.LOCAL_ONLY, 0)
self.add_dataset_to_json_file(current_dataset, self.dataset_list_path)
# Connect the robot and teleoperator # Connect the robot and teleoperator
if not self.robot.is_connected: if not self.robot.is_connected: