Safran/KPConv-PyTorch
1
0
Fork 0
Code Issues Pull requests Packages Projects Releases Wiki Activity

.

Browse source
This commit is contained in:
HuguesTHOMAS 2022-04-04 17:34:53 -04:00
parent e600c1667d
commit 9a18c66ba8
1 changed files with 6 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
datasets/S3DIS.py
View file

@ -340,6 +340,12 @@ class S3DISDataset(PointCloudDataset):
# Number collected # Number collected
n = input_inds.shape[0] n = input_inds.shape[0]
# Safe check for empty spheres
if n < 2:
t += [time.time()]
t += [time.time()]
continue
# Collect labels and colors # Collect labels and colors
input_points = (points[input_inds] - center_point).astype(np.float32) input_points = (points[input_inds] - center_point).astype(np.float32)
input_colors = self.input_colors[cloud_ind][input_inds] input_colors = self.input_colors[cloud_ind][input_inds]