Publish 'RoboAiUR5e' OXE datasets, see https://robotics-transformer-x.github.io/.

PiperOrigin-RevId: 907039327

Author: The TensorFlow Datasets Authors

tensorflow_datasets/robotics/__init__.py

@@ -66,6 +66,7 @@
 from tensorflow_datasets.robotics.rtx import NyuFrankaPlayDatasetConvertedExternallyToRlds
 from tensorflow_datasets.robotics.rtx import NyuRotDatasetConvertedExternallyToRlds
 from tensorflow_datasets.robotics.rtx import PlexRobosuite
+from tensorflow_datasets.robotics.rtx import RoboAiUR5e
 from tensorflow_datasets.robotics.rtx import RoboSet
 from tensorflow_datasets.robotics.rtx import Roboturk
 from tensorflow_datasets.robotics.rtx import SpocRobot

tensorflow_datasets/robotics/rtx/__init__.py

@@ -55,6 +55,7 @@
 from tensorflow_datasets.robotics.rtx.rtx import NyuFrankaPlayDatasetConvertedExternallyToRlds
 from tensorflow_datasets.robotics.rtx.rtx import NyuRotDatasetConvertedExternallyToRlds
 from tensorflow_datasets.robotics.rtx.rtx import PlexRobosuite
+from tensorflow_datasets.robotics.rtx.rtx import RoboAiUR5e
 from tensorflow_datasets.robotics.rtx.rtx import RoboSet
 from tensorflow_datasets.robotics.rtx.rtx import Roboturk
 from tensorflow_datasets.robotics.rtx.rtx import SpocRobot

tensorflow_datasets/robotics/rtx/rtx.py

@@ -1432,3 +1432,25 @@ def get_homepage(self):
 
   def get_relative_dataset_location(self):
     return 'spoc/0.0.1'
+
+
+class RoboAiUR5e(dataset_importer_builder.DatasetImporterBuilder):
+  """DatasetBuilder for `robo_ai_u_r5e` dataset."""
+
+  def get_description(self):
+    return r"""Universal Robots UR5e demonstration dataset for VLA model training. Data collected from real-world robot workspace with preprogrammed robot routines."""
+
+  def get_citation(self):
+    return r"""@article{RoboAIUR5e2025,
+  title        = {RoboAI UR5e Training Dataset},
+  author       = {Joonas Rouhiainen},
+  institution  = {RoboAI Research Center, Satakunta University of Applied Sciences},
+  year         = {2025},
+  howpublished = {\url{https://sites.google.com/view/roboai-ur5e/home}},
+}"""
+
+  def get_homepage(self):
+    return r"""https://sites.google.com/view/roboai-ur5e/home"""
+
+  def get_relative_dataset_location(self):
+    return 'robo_ai_u_r5e/0.0.1'