From 9dfba7ecd8e29c0a0de9c71bc9d8a0e0d16e8695 Mon Sep 17 00:00:00 2001
From: Kosuke Takeuchi <kosuke.tnp@gmail.com>
Date: Thu, 9 Jul 2020 05:53:58 +0900
Subject: [PATCH] Add mug to synset_to_label of shapenet

Signed-off-by: Kosuke Takeuchi <kosuke.tnp@gmail.com>
---
 kaolin/datasets/shapenet.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/kaolin/datasets/shapenet.py b/kaolin/datasets/shapenet.py
index 1646f9ce6..97b9fc59b 100644
--- 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()}