fix round(np.float64) to be int

open-mmlab · Jun 30, 2021 · 793bcb0 · 793bcb0
1 parent d0d3d94
commit 793bcb0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tools/data_converter/scannet_data_utils.py b/tools/data_converter/scannet_data_utils.py
@@ -244,7 +244,7 @@ def get_scene_idxs_and_label_weight(self):
         num_iter = int(np.sum(num_point_all) / float(self.num_points))
         scene_idxs = []
         for idx in range(len(self.data_infos)):
-            scene_idxs.extend([idx] * round(sample_prob[idx] * num_iter))
+            scene_idxs.extend([idx] * int(round(sample_prob[idx] * num_iter)))
         scene_idxs = np.array(scene_idxs).astype(np.int32)
 
         # calculate label weight, adopted from PointNet++