This commit is contained in:
HuguesTHOMAS 2021-07-29 22:35:33 +00:00
parent 7f5f52b067
commit a223db4d65

View file

@ -290,7 +290,7 @@ class PointCloudDataset(Dataset):
if self.config.augment_scale_anisotropic:
scale = np.random.rand(points.shape[1]) * (max_s - min_s) + min_s
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
symmetries = np.array(self.config.augment_symmetries).astype(np.int32)