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:43:20 -04:00
parent 9a18c66ba8
commit 5b5641e02d
1 changed files with 12 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
datasets/S3DIS.py
View file

@ -253,6 +253,7 @@ class S3DISDataset(PointCloudDataset):
s_list = [] s_list = []
R_list = [] R_list = []
batch_n = 0 batch_n = 0
failed_attempts = 0
info = get_worker_info() info = get_worker_info()
if info is not None: if info is not None:
@ -342,6 +343,9 @@ class S3DISDataset(PointCloudDataset):
# Safe check for empty spheres # Safe check for empty spheres
if n < 2: if n < 2:
failed_attempts += 1
if failed_attempts > 2 * self.config.batch_num:
raise ValueError('It seems this dataset only containes empty input spheres')
t += [time.time()] t += [time.time()]
t += [time.time()] t += [time.time()]
continue continue
@ -510,6 +514,7 @@ class S3DISDataset(PointCloudDataset):
s_list = [] s_list = []
R_list = [] R_list = []
batch_n = 0 batch_n = 0
failed_attempts = 0
while True: while True:
@ -542,6 +547,13 @@ 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:
failed_attempts += 1
if failed_attempts > 2 * self.config.batch_num:
raise ValueError('It seems this dataset only containes empty input spheres')
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]