fixing bug in random_sampling for NPM3D

This commit is contained in:
meyerjo 2023-04-24 13:54:30 +02:00
parent 1defcd75cf
commit b2af237e31

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]