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

failed attempt correction

Browse source
This commit is contained in:
HuguesTHOMAS 2022-04-06 12:35:19 -04:00
parent 5b5641e02d
commit 7255680ff0
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
datasets/S3DIS.py
View file

@ -344,7 +344,7 @@ class S3DISDataset(PointCloudDataset):
# Safe check for empty spheres # Safe check for empty spheres
if n < 2: if n < 2:
failed_attempts += 1 failed_attempts += 1
if failed_attempts > 2 * self.config.batch_num: if failed_attempts > 100 * self.config.batch_num:
raise ValueError('It seems this dataset only containes empty input spheres') raise ValueError('It seems this dataset only containes empty input spheres')
t += [time.time()] t += [time.time()]
t += [time.time()] t += [time.time()]
@ -550,7 +550,7 @@ class S3DISDataset(PointCloudDataset):
# Safe check for empty spheres # Safe check for empty spheres
if n < 2: if n < 2:
failed_attempts += 1 failed_attempts += 1
if failed_attempts > 2 * self.config.batch_num: if failed_attempts > 100 * self.config.batch_num:
raise ValueError('It seems this dataset only containes empty input spheres') raise ValueError('It seems this dataset only containes empty input spheres')
continue continue