semseg int64 bug

magicleap · Sep 24, 2020 · 5c55a5a · 5c55a5a
1 parent bee3ddb
commit 5c55a5a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/atlas/tsdf.py b/atlas/tsdf.py
@@ -193,13 +193,13 @@ def get_mesh(self, attribute='color', cmap='nyu40'):
         if 'semseg' in self.attribute_vols:
             semseg_vol = self.attribute_vols['semseg'].detach().cpu().numpy()
             semseg = semseg_vol[verts_ind[:,0],verts_ind[:,1],verts_ind[:,2]]
-            vertex_attributes['semseg'] = semseg
+            vertex_attributes['semseg'] = semseg.astype(np.int32)
 
         if 'instance' in self.attribute_vols:
             instance_vol = self.attribute_vols['instance']
             instance_vol = instance_vol.detach().cpu().numpy()
             instance = instance_vol[verts_ind[:,0],verts_ind[:,1],verts_ind[:,2]]
-            vertex_attributes['instance'] = instance
+            vertex_attributes['instance'] = instance.astype(np.int32)
 
         # color mesh
         if attribute=='color' and 'color' in self.attribute_vols: