Merge pull request #216 from meyerjo/fix_random_sampling

Bug in NPM3d dataloader
This commit is contained in:
Hugues THOMAS 2023-04-24 16:00:28 -04:00 committed by GitHub
commit d5391238e5
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -580,7 +580,7 @@ class NPM3DDataset(PointCloudDataset):
# input_colors *= 0
# Get original height as additional feature
input_features = np.hstack((input_colors, input_points[:, 2:] + center_point[:, 2:])).astype(np.float32)
input_features = np.hstack((input_points[:, 2:] + center_point[:, 2:])).astype(np.float32)
# Stack batch
p_list += [input_points]