Skip to content
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

Generic approach for capturing datasets #141

Open
fred-labs opened this issue Jul 31, 2024 · 0 comments
Open

Generic approach for capturing datasets #141

fred-labs opened this issue Jul 31, 2024 · 0 comments

Comments

@fred-labs
Copy link
Contributor

Currently, the only supported method for capturing data with scenario execution is using ROS2 (using the action record_bag).

We want to support at least one other simulation, e.g. pybullet and provide a (generic) approach to capture data.

Example:

import osc.helpers
import osc.pybullet
import osc.data_writer # <-- new library

scenario example_pybullet:
    timeout(60s)
    sim: simulation_pybullet
    actor1: actor_pybullet
    writer: kitti_writer("./my_dataset", "odometry1") # <-- new actor
    do serial:
        sim.initialize("plane.urdf")
        actor1.spawn("husky/husky.urdf", pose_3d(position: position_3d(z: 0.5)))
        parallel:
            sim.run()
            writer.capture() # <-- new action
            actor1.set_joint_motor_control(5mps, 1)
            serial:
                actor1.distance_traveled(2m)
                emit end

In this example only the odometry might be captured (as no image provider is instantiated).

Different data annotation formats might be supported, e.g.:

  • kitti
  • csv/jpg
  • ...

Stretch Goal

Create example on how to use captured dataset to train an AI.

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

No branches or pull requests

1 participant