diff --git a/demo.py b/demo.py index ed545d65a..f14481159 100644 --- a/demo.py +++ b/demo.py @@ -5,7 +5,6 @@ """ Demo of using VoteNet 3D object detector to detect objects from a point cloud. """ - import os import sys import numpy as np diff --git a/scannet/scannet_detection_dataset.py b/scannet/scannet_detection_dataset.py index eaca9c906..3df68aa05 100644 --- a/scannet/scannet_detection_dataset.py +++ b/scannet/scannet_detection_dataset.py @@ -75,10 +75,10 @@ def __getitem__(self, idx): """ scan_name = self.scan_names[idx] - mesh_vertices = np.load(os.path.join(self.data_path, scan_name)+'_vert.npy') - instance_labels = np.load(os.path.join(self.data_path, scan_name)+'_ins_label.npy') - semantic_labels = np.load(os.path.join(self.data_path, scan_name)+'_sem_label.npy') - instance_bboxes = np.load(os.path.join(self.data_path, scan_name)+'_bbox.npy') + mesh_vertices = np.load(os.path.join(self.data_path, scan_name)+'_vert.npy', allow_pickle=True) + instance_labels = np.load(os.path.join(self.data_path, scan_name)+'_ins_label.npy', allow_pickle=True) + semantic_labels = np.load(os.path.join(self.data_path, scan_name)+'_sem_label.npy', allow_pickle=True) + instance_bboxes = np.load(os.path.join(self.data_path, scan_name)+'_bbox.npy', allow_pickle=True) if not self.use_color: point_cloud = mesh_vertices[:,0:3] # do not use color for now