toddlerbot.manipulation.datasets package¶
Submodules¶
toddlerbot.manipulation.datasets.get_pose_data module¶
- toddlerbot.manipulation.datasets.get_pose_data.process_raw_dataset(robot: Robot, task: str, time_str: str, dt: float = 0.1, time_offset: float = 0.2)¶
Processes raw dataset files for a specified robot and task, extracting and saving relevant data.
- Parameters:
robot (Robot) – The robot instance containing joint limits and motor ordering.
task (str) – The task type, either “pick” or another task, determining data extraction logic.
time_str (str) – A string representing the time, used to locate the dataset directory.
dt (float, optional) – The time step for data processing. Defaults to 0.1.
time_offset (float, optional) – The time offset applied to the data. Defaults to 0.2.
Processes files in the specified dataset directory, extracting time, action, and motor position data. Concatenates and saves the processed data into a new dataset file, and generates plots for visualization.
toddlerbot.manipulation.datasets.process_raw_data module¶
- toddlerbot.manipulation.datasets.process_raw_data.process_raw_dataset(robot: Robot, task: str, time_str: str, dt: float = 0.1, time_offset: float = 0.2)¶
Processes raw dataset files for a specified robot and task, resampling and organizing data for further analysis.
- Parameters:
robot (Robot) – The robot object containing configuration and state information.
task (str) – The task type, which determines how actions and positions are processed.
time_str (str) – A string representing the time, used for file path generation.
dt (float, optional) – The time interval for resampling the data. Defaults to 0.1.
time_offset (float, optional) – The time offset applied during resampling. Defaults to 0.2.