Unverified Commit 52ca8987 authored by wawatt's avatar wawatt Committed by GitHub
Browse files

Update my_detection_dataset

parent e41d2dd5
......@@ -48,7 +48,7 @@ class myDetectionDataset(Dataset):
# ------------------------------- LABELS ------------------------------
ret_dict = {}
point_cloud = pc_util.random_sampling(point_cloud, self.num_points)
one_hot = DC.sem2class(bbox[19])
one_hot = DC.sem2class(bbox[16])
one_hot_ex_rep = np.repeat(np.expand_dims(one_hot, axis=0), self.num_points, axis=0)
point_cloud_with_cls = np.concatenate((point_cloud, one_hot_ex_rep), axis=1)
......@@ -64,4 +64,3 @@ class myDetectionDataset(Dataset):
return ret_dict
Supports Markdown
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