Skip to content

GitLab

Unverified Commit 18fec567 authored by Danila Rukhovich's avatar Danila Rukhovich Committed by GitHub
Browse files

fix round(np.float64) to be int (#687)

parent f5859d1b
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tools/data_converter/scannet_data_utils.py
View file @ 18fec567
...@@ -244,7 +244,7 @@ class ScanNetSegData(object): ...@@ -244,7 +244,7 @@ class ScanNetSegData(object):
num_iter = int(np.sum(num_point_all) / float(self.num_points)) num_iter = int(np.sum(num_point_all) / float(self.num_points))
scene_idxs = [] scene_idxs = []
for idx in range(len(self.data_infos)): 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) scene_idxs = np.array(scene_idxs).astype(np.int32)
# calculate label weight, adopted from PointNet++ # calculate label weight, adopted from PointNet++
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment