Skip to content

Missing GroundingTruth entries for selected google_robot_* tasks in Fractal dataset #106

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
MstarLioning opened this issue May 29, 2025 · 1 comment

Comments

@MstarLioning
Copy link

Hello,We have selected the following tasks in SImplerenv:

[
    "google_robot_pick_coke_can",
    "google_robot_pick_horizontal_coke_can", 
    "google_robot_pick_vertical_coke_can",
    "google_robot_pick_standing_coke_can",
    "google_robot_pick_object",
    "google_robot_move_near_v0",
    "google_robot_move_near_v1",
    "google_robot_move_near"
]

We’d like to locate the corresponding episodes (and GroundingTruth annotations) for each of these tasks in the Fractal dataset. However, after searching through the first 15,000 episodes, we only found a GroundingTruth entry for google_robot_pick_coke_can. Is this expected? Have we overlooked something?

Our extraction method is as follows:

# Assign episode_id and extract
episode_data = self._extract_episode_data(episode, matched_task, task_counters[matched_task])

# In _extract_episode_data we assemble the actual action trajectory:
action_trajectory = np.array([
    np.concatenate([
        step['action']['world_vector'].numpy(),
        step['action']['rotation_delta'].numpy(),
        step['action']['gripper_closedness_action'].numpy()
    ])
    for step in episode['steps']
])

Any pointers on whether the other tasks should appear under different names, or if there’s an alternative way to locate their GroundingTruth, would be greatly appreciated. Thank you!

@xuanlinli17
Copy link
Collaborator

Could you elaborate what "ground truth" means? The names of the mp4 files saved for visualization have the success conditions.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants