.
This commit is contained in:
parent
e600c1667d
commit
9a18c66ba8
|
@ -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]
|
||||||
|
|
Loading…
Reference in a new issue