add mp3d_example scene dataset config to the tutorials (#1847)

facebookresearch · Aug 30, 2022 · a0061df · a0061df
1 parent 113c3f1
commit a0061df
Show file tree

Hide file tree

Showing 6 changed files with 10 additions and 0 deletions.
diff --git a/examples/tutorials/colabs/ECCV_2020_Advanced_Features.ipynb b/examples/tutorials/colabs/ECCV_2020_Advanced_Features.ipynb
@@ -513,6 +513,7 @@
     "        \"width\": 720,  # Spatial resolution of the observations\n",
     "        \"height\": 544,\n",
     "        \"scene\": \"./data/scene_datasets/mp3d_example/17DRP5sb8fy/17DRP5sb8fy.glb\",  # Scene path\n",
+    "        \"scene_dataset\": \"./data/scene_datasets/mp3d_example/mp3d.scene_dataset_config.json\",  # mp3d scene dataset\n",
     "        \"default_agent\": 0,\n",
     "        \"sensor_height\": 1.5,  # Height of sensors in meters\n",
     "        \"sensor_pitch\": -math.pi / 8.0,  # sensor pitch (x rotation in rads)\n",

diff --git a/examples/tutorials/colabs/ECCV_2020_Interactivity.ipynb b/examples/tutorials/colabs/ECCV_2020_Interactivity.ipynb
@@ -212,6 +212,7 @@
     "        \"width\": 720,  # Spatial resolution of the observations\n",
     "        \"height\": 544,\n",
     "        \"scene\": \"./data/scene_datasets/mp3d_example/17DRP5sb8fy/17DRP5sb8fy.glb\",  # Scene path\n",
+    "        \"scene_dataset_config\": \"./data/scene_datasets/mp3d_example/mp3d.scene_dataset_config.json\",  # MP3D scene dataset\n",
     "        \"default_agent\": 0,\n",
     "        \"sensor_height\": 1.5,  # Height of sensors in meters\n",
     "        \"sensor_pitch\": -math.pi / 8.0,  # sensor pitch (x rotation in rads)\n",

diff --git a/examples/tutorials/colabs/ECCV_2020_Navigation.ipynb b/examples/tutorials/colabs/ECCV_2020_Navigation.ipynb
@@ -352,6 +352,7 @@
     "# @title Configure Sim Settings\n",
     "\n",
     "test_scene = \"./data/scene_datasets/mp3d_example/17DRP5sb8fy/17DRP5sb8fy.glb\"\n",
+    "mp3d_scene_dataset = \"./data/scene_datasets/mp3d_example/mp3d.scene_dataset_config.json\"\n",
     "\n",
     "rgb_sensor = True  # @param {type:\"boolean\"}\n",
     "depth_sensor = True  # @param {type:\"boolean\"}\n",
@@ -361,6 +362,7 @@
     "    \"width\": 256,  # Spatial resolution of the observations\n",
     "    \"height\": 256,\n",
     "    \"scene\": test_scene,  # Scene path\n",
+    "    \"scene_dataset\": mp3d_scene_dataset,  # the scene dataset configuration files\n",
     "    \"default_agent\": 0,\n",
     "    \"sensor_height\": 1.5,  # Height of sensors in meters\n",
     "    \"color_sensor\": rgb_sensor,  # RGB sensor\n",
@@ -381,6 +383,7 @@
     "    sim_cfg = habitat_sim.SimulatorConfiguration()\n",
     "    sim_cfg.gpu_device_id = 0\n",
     "    sim_cfg.scene_id = settings[\"scene\"]\n",
+    "    sim_cfg.scene_dataset_config_file = settings[\"scene_dataset\"]\n",
     "    sim_cfg.enable_physics = settings[\"enable_physics\"]\n",
     "\n",
     "    # Note: all sensors must have the same resolution\n",

diff --git a/examples/tutorials/nb_python/ECCV_2020_Advanced_Features.py b/examples/tutorials/nb_python/ECCV_2020_Advanced_Features.py
@@ -504,6 +504,7 @@ def make_default_settings():
         "width": 720,  # Spatial resolution of the observations
         "height": 544,
         "scene": "./data/scene_datasets/mp3d_example/17DRP5sb8fy/17DRP5sb8fy.glb",  # Scene path
+        "scene_dataset": "./data/scene_datasets/mp3d_example/mp3d.scene_dataset_config.json",  # mp3d scene dataset
         "default_agent": 0,
         "sensor_height": 1.5,  # Height of sensors in meters
         "sensor_pitch": -math.pi / 8.0,  # sensor pitch (x rotation in rads)

diff --git a/examples/tutorials/nb_python/ECCV_2020_Interactivity.py b/examples/tutorials/nb_python/ECCV_2020_Interactivity.py
@@ -203,6 +203,7 @@ def make_default_settings():
         "width": 720,  # Spatial resolution of the observations
         "height": 544,
         "scene": "./data/scene_datasets/mp3d_example/17DRP5sb8fy/17DRP5sb8fy.glb",  # Scene path
+        "scene_dataset_config": "./data/scene_datasets/mp3d_example/mp3d.scene_dataset_config.json",  # MP3D scene dataset
         "default_agent": 0,
         "sensor_height": 1.5,  # Height of sensors in meters
         "sensor_pitch": -math.pi / 8.0,  # sensor pitch (x rotation in rads)

diff --git a/examples/tutorials/nb_python/ECCV_2020_Navigation.py b/examples/tutorials/nb_python/ECCV_2020_Navigation.py
@@ -282,6 +282,7 @@ def navigateAndSee(action=""):
 # @title Configure Sim Settings
 
 test_scene = "./data/scene_datasets/mp3d_example/17DRP5sb8fy/17DRP5sb8fy.glb"
+mp3d_scene_dataset = "./data/scene_datasets/mp3d_example/mp3d.scene_dataset_config.json"
 
 rgb_sensor = True  # @param {type:"boolean"}
 depth_sensor = True  # @param {type:"boolean"}
@@ -291,6 +292,7 @@ def navigateAndSee(action=""):
     "width": 256,  # Spatial resolution of the observations
     "height": 256,
     "scene": test_scene,  # Scene path
+    "scene_dataset": mp3d_scene_dataset,  # the scene dataset configuration files
     "default_agent": 0,
     "sensor_height": 1.5,  # Height of sensors in meters
     "color_sensor": rgb_sensor,  # RGB sensor
@@ -306,6 +308,7 @@ def make_cfg(settings):
     sim_cfg = habitat_sim.SimulatorConfiguration()
     sim_cfg.gpu_device_id = 0
     sim_cfg.scene_id = settings["scene"]
+    sim_cfg.scene_dataset_config_file = settings["scene_dataset"]
     sim_cfg.enable_physics = settings["enable_physics"]
 
     # Note: all sensors must have the same resolution