Add mug to synset_to_label of shapenet

Signed-off-by: Kosuke Takeuchi <[email protected]>
NVIDIAGameWorks · Jul 9, 2020 · 9dfba7e · 9dfba7e
1 parent ec193ff
commit 9dfba7e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/kaolin/datasets/shapenet.py b/kaolin/datasets/shapenet.py
@@ -56,7 +56,8 @@
                    '03642806': 'laptop', '03710193': 'mailbox', '03761084': 'microwave',
                    '03928116': 'piano', '03938244': 'pillow', '03948459': 'pistol',
                    '04004475': 'printer', '04099429': 'rocket', '04256520': 'sofa',
-                   '04554684': 'washer', '04090263': 'rifle', '02946921': 'can'}
+                   '04554684': 'washer', '04090263': 'rifle', '02946921': 'can',
+                   '03797390': 'mug'}
 
 # Label to Synset mapping (for ShapeNet core classes)
 label_to_synset = {v: k for k, v in synset_to_label.items()}