updating aliases for MP3D object names

bdaiinstitute · Sep 1, 2023 · 030bbe7 · 030bbe7
1 parent fd866a9
commit 030bbe7
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 12 deletions.
diff --git a/zsos/policy/habitat_policies.py b/zsos/policy/habitat_policies.py
@@ -26,13 +26,13 @@
 HM3D_ID_TO_NAME = ["chair", "bed", "potted plant", "toilet", "tv", "couch"]
 MP3D_ID_TO_NAME = [
     "chair",
-    "table",
+    "dining table",  # "table",
     "picture",
     "cabinet",
-    "cushion",
+    "pillow",  # "cushion",
     "couch",  # "sofa",
     "bed",
-    "chest of drawers",
+    "dresser",  # "chest of drawers",
     "potted plant",  # "plant",
     "sink",
     "toilet",

diff --git a/zsos/vlm/classes.txt b/zsos/vlm/classes.txt
@@ -1,17 +1,10 @@
-chair
-table
 picture
 cabinet
-cushion
-couch
-bed
-chest of drawers
-potted plant
+pillow
+dresser
 sink
-toilet
 stool
 towel
-tv
 shower
 bathtub
 counter