Ecosyste.ms: Awesome

An open API service indexing awesome lists of open source software.

Awesome Lists | Featured Topics | Projects

https://github.com/unity-technologies/robotics-object-pose-estimation

A complete end-to-end demonstration in which we collect training data in Unity and use that data to train a deep neural network to predict the pose of a cube. This model is then deployed in a simulated robotic pick-and-place task.
https://github.com/unity-technologies/robotics-object-pose-estimation

autonomy computer-vision deep-learning machine-learning manipulation model-training motion-planning perception physics-simulation pose-estimation robotics robotics-simulation ros simulation synthetic-data trajectory-generation tutorial unity ur3-robot-arm urdf

Last synced: 6 days ago
JSON representation

A complete end-to-end demonstration in which we collect training data in Unity and use that data to train a deep neural network to predict the pose of a cube. This model is then deployed in a simulated robotic pick-and-place task.

Awesome Lists containing this project

README 

        
# Object Pose Estimation Demo



[![License](https://img.shields.io/badge/License-Apache%202.0-blue.svg)](https://opensource.org/licenses/Apache-2.0)



This tutorial will go through the steps necessary to perform pose estimation with a UR3 robotic arm in Unity. You’ll gain experience integrating ROS with Unity, importing URDF models, collecting labeled training data, and training and deploying a deep learning model. By the end of this tutorial, you will be able to perform pick-and-place with a robot arm in Unity, using computer vision to perceive the object the robot picks up.



**Want to skip the tutorial and run the full demo? Check out our [Quick Demo](Documentation/quick_demo_full.md).**



**Want to skip the tutorial and focus on collecting training data for the deep learning model? Check out our [Quick Data-Collection Demo](Documentation/quick_demo_train.md).**



> Note: This project has been developed with Python 3 and ROS Noetic.



**Table of Contents**

- [Part 1: Create Unity scene with imported URDF](#link-part-1)

- [Part 2: Setting up the scene for data collection](#link-part-2)

- [Part 3: Data Collection and Model Training](#link-part-3)

- [Part 4: Pick-and-Place](#link-part-4)



---



### [Part 1: Create Unity Scene with Imported URDF](Documentation/1_set_up_the_scene.md)






This part includes downloading and installing the Unity Editor, setting up a basic Unity scene, and importing a robot. We will import the [UR3 robot arm](https://www.universal-robots.com/products/ur3-robot) using the [URDF Importer](https://github.com/Unity-Technologies/URDF-Importer) package.



---



### [Part 2: Setup the Scene for Data Collection](Documentation/2_set_up_the_data_collection_scene.md)






This part focuses on setting up the scene for data collection using the Unity Computer Vision [Perception Package](https://github.com/Unity-Technologies/com.unity.perception). You will learn how to use Perception Package [Randomizers](https://github.com/Unity-Technologies/com.unity.perception/blob/master/com.unity.perception/Documentation~/Randomization/Index.md) to randomize aspects of the scene in order to create variety in the training data.



If you would like to learn more about Randomizers, and apply domain randomization to this scene more thoroughly, check out our further exercises for the reader [here](Documentation/5_more_randomizers.md).



---



### [Part 3: Data Collection and Model Training](Documentation/3_data_collection_model_training.md)






This part includes running data collection with the Perception Package, and using that data to train a deep learning model. The training step can take some time. If you'd like, you can skip that step by using our pre-trained model.



To measure the success of grasping in simulation using our pre-trained model for pose estimation, we did 100 trials and got the following results:



|                  | Success | Failures | Percent Success |

|:----------------:|:-------:|:--------:|:---------------:|

|Without occlusion |    82   |     5    |      94         |

|With occlusion    |    7    |     6    |      54         |

|All               |    89   |     11   |      89         |



> Note: Data for the above experiment was collected in Unity 2020.2.1f1.



---



### [Part 4: Pick-and-Place](Documentation/4_pick_and_place.md)






This part includes the preparation and setup necessary to run a pick-and-place task using MoveIt. Here, the cube pose is predicted by the trained deep learning model. Steps covered include:

* Creating and invoking a motion planning service in ROS

* Sending captured RGB images from our scene to the ROS Pose Estimation node for inference

* Using a Python script to run inference on our trained deep learning model

* Moving Unity Articulation Bodies based on a calculated trajectory

* Controlling a gripping tool to successfully grasp and drop an object.



---



## Support

For questions or discussions about Unity Robotics package installations or how to best set up and integrate your robotics projects, please create a new thread on the [Unity Robotics forum](https://forum.unity.com/forums/robotics.623/) and make sure to include as much detail as possible.



For feature requests, bugs, or other issues, please file a [GitHub issue](https://github.com/Unity-Technologies/Robotics-Object-Pose-Estimation/issues) using the provided templates and the Robotics team will investigate as soon as possible.



For any other questions or feedback, connect directly with the

Robotics team at [[email protected]](mailto:[email protected]).



## More from Unity Robotics

Visit the [Robotics Hub](https://github.com/Unity-Technologies/Unity-Robotics-Hub) for more tutorials, tools, and information on robotics simulation in Unity!



## License

[Apache License 2.0](LICENSE)