Update graspnet_dataset.py

xiaomengfang · Jun 27, 2023 · 50dd448 · 50dd448
1 parent d4a52af
commit 50dd448
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/dataset/graspnet_dataset.py b/dataset/graspnet_dataset.py
@@ -237,6 +237,7 @@ def load_grasp_labels(root):
 def minkowski_collate_fn(list_data):
     coordinates_batch, features_batch = ME.utils.sparse_collate([d["coors"] for d in list_data],
                                                                 [d["feats"] for d in list_data])
+    coordinates_batch = np.ascontiguousarray(coordinates_batch, dtype=np.int32)
     coordinates_batch, features_batch, _, quantize2original = ME.utils.sparse_quantize(
         coordinates_batch, features_batch, return_index=True, return_inverse=True)
     res = {