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

.

Browse source
This commit is contained in:
HuguesTHOMAS 2021-07-29 22:35:33 +00:00
parent 7f5f52b067
commit a223db4d65
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
datasets/common.py
View file

@ -290,7 +290,7 @@ class PointCloudDataset(Dataset):
if self.config.augment_scale_anisotropic: if self.config.augment_scale_anisotropic:
scale = np.random.rand(points.shape[1]) * (max_s - min_s) + min_s scale = np.random.rand(points.shape[1]) * (max_s - min_s) + min_s
else: else:
scale = np.random.rand() * (max_s - min_s) - min_s scale = np.random.rand() * (max_s - min_s) + min_s
# Add random symmetries to the scale factor # Add random symmetries to the scale factor
symmetries = np.array(self.config.augment_symmetries).astype(np.int32) symmetries = np.array(self.config.augment_symmetries).astype(np.int32)