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

PiperOrigin-RevId: 627758161

Author: The TensorFlow Datasets Authors

tensorflow_datasets/robotics/__init__.py

@@ -54,8 +54,10 @@
 from tensorflow_datasets.robotics.rtx import NyuDoorOpeningSurprisingEffectiveness
 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 RoboSet
 from tensorflow_datasets.robotics.rtx import Roboturk
+from tensorflow_datasets.robotics.rtx import Spoc
 from tensorflow_datasets.robotics.rtx import StanfordHydraDatasetConvertedExternallyToRlds
 from tensorflow_datasets.robotics.rtx import StanfordKukaMultimodalDatasetConvertedExternallyToRlds
 from tensorflow_datasets.robotics.rtx import StanfordMaskVitConvertedExternallyToRlds

tensorflow_datasets/robotics/rtx/__init__.py

@@ -54,8 +54,10 @@
 from tensorflow_datasets.robotics.rtx.rtx import NyuDoorOpeningSurprisingEffectiveness
 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 RoboSet
 from tensorflow_datasets.robotics.rtx.rtx import Roboturk
+from tensorflow_datasets.robotics.rtx.rtx import Spoc
 from tensorflow_datasets.robotics.rtx.rtx import StanfordHydraDatasetConvertedExternallyToRlds
 from tensorflow_datasets.robotics.rtx.rtx import StanfordKukaMultimodalDatasetConvertedExternallyToRlds
 from tensorflow_datasets.robotics.rtx.rtx import StanfordMaskVitConvertedExternallyToRlds

tensorflow_datasets/robotics/rtx/rtx.py

@@ -1390,3 +1390,45 @@ def get_homepage(self):
 
   def get_relative_dataset_location(self):
     return 'bridge_data_msr/0.0.1'
+
+
+class PlexRobosuite(dataset_importer_builder.DatasetImporterBuilder):
+  """DatasetBuilder for `plex_robosuite` dataset."""
+
+  def get_description(self):
+    return (
+        "A dataset of high-qualty demonstration trajectories for Robosuite's"
+        ' Door, Stack, PickPlaceMilk, PickPlaceBread, PickPlaceCereal, and'
+        ' NutAssemblyRound tasks, 75 demonstrations per each.'
+    )
+
+  def get_citation(self):
+    return 'https://doi.org/10.48550/arXiv.2303.08789'
+
+  def get_homepage(self):
+    return 'https://microsoft.github.io/PLEX/'
+
+  def get_relative_dataset_location(self):
+    return 'plex_robosuite/0.0.1'
+
+
+class Spoc(dataset_importer_builder.DatasetImporterBuilder):
+  """DatasetBuilder for `spoc` dataset."""
+
+  def get_description(self):
+    return ''
+
+  def get_citation(self):
+    return r"""@article{spoc2023,
+    author    = {Kiana Ehsani, Tanmay Gupta, Rose Hendrix, Jordi Salvador, Luca Weihs, Kuo-Hao Zeng, Kunal Pratap Singh, Yejin Kim, Winson Han, Alvaro Herrasti, Ranjay Krishna, Dustin Schwenk, Eli VanderBilt, Aniruddha Kembhavi},
+    title     = {Imitating Shortest Paths in Simulation Enables Effective Navigation and Manipulation in the Real World},
+    journal   = {arXiv},
+    year      = {2023},
+    eprint    = {2312.02976},
+}"""
+
+  def get_homepage(self):
+    return 'https://spoc-robot.github.io/'
+
+  def get_relative_dataset_location(self):
+    return 'spoc/0.0.1'