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-07-11 10:35:17 -04:00
parent 843629422b
commit 7d4c03d199
1 changed files with 2 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
datasets/SemanticKitti.py
View file

@ -230,6 +230,8 @@ class SemanticKittiDataset(PointCloudDataset):
# Update epoch indice # Update epoch indice
self.epoch_i += 1 self.epoch_i += 1
if self.epoch_i >= int(self.epoch_inds.shape[0]):
self.epoch_i -= int(self.epoch_inds.shape[0])
s_ind, f_ind = self.all_inds[ind] s_ind, f_ind = self.all_inds[ind]