Skip to content

GitLab

Commit 60e019e3 authored by Shaoshuai Shi's avatar Shaoshuai Shi
Browse files

small bug in waymo_eval.py

parent 25d9f503
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
pcdet/datasets/waymo/waymo_eval.py
View file @ 60e019e3
...@@ -71,7 +71,7 @@ class OpenPCDetWaymoDetectionMetricsEstimator(tf.test.TestCase): ...@@ -71,7 +71,7 @@ class OpenPCDetWaymoDetectionMetricsEstimator(tf.test.TestCase):
boxes3d.append(np.array(info['boxes_lidar'])) boxes3d.append(np.array(info['boxes_lidar']))
box_name = info['name'] box_name = info['name']
if boxes3d[-1].shape[-1] == 9: if boxes3d[-1].shape[-1] == 9:
boxes3d[-1] = boxes3d[-1][", 0:7"] boxes3d[-1] = boxes3d[-1][:, 0:7]
obj_type += [self.WAYMO_CLASSES.index(name) for i, name in enumerate(box_name)] obj_type += [self.WAYMO_CLASSES.index(name) for i, name in enumerate(box_name)]
frame_id.append(np.array([frame_index] * num_boxes)) frame_id.append(np.array([frame_index] * num_boxes))
......
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