add category names, fix page link

ajzhai · Sep 5, 2023 · 5fe8358 · 5fe8358
1 parent a69c404
commit 5fe8358
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/README.md b/README.md
@@ -5,7 +5,7 @@ University of Illinois at Urbana-Champaign
 
 ICCV 2023
 
-[Paper](https://arxiv.org/abs/2212.02497) │ [Project Page](https://ajzhai.github.io/peanut/)
+[Paper](https://arxiv.org/abs/2212.02497) │ [Project Page](https://ajzhai.github.io/PEANUT/)
 
 ![Example Video](docs/example_vid.gif)
 

diff --git a/nav/constants.py b/nav/constants.py
@@ -29,6 +29,8 @@
                 5: 1}
 coco_to_hm3d = {v : k for k, v in hm3d_to_coco.items()}
 
+map_category_names = {0: "chair", 1: "sofa", 2: "plant",  3: "bed",  4: "toilet",  5: "tv_monitor",
+                      6: "fireplace", 7: "bathtub", 8: "mirror"}
 
 coco_categories_mapping = {
     56: 0,  # chair