From 5c55a5a291b0eb988ace035a0b1283ff52e02c5d Mon Sep 17 00:00:00 2001 From: zmurez Date: Thu, 24 Sep 2020 10:36:44 -0700 Subject: [PATCH] semseg int64 bug --- atlas/tsdf.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/atlas/tsdf.py b/atlas/tsdf.py index bbbdb5c..dec87ac 100644 --- 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: