Skip to content

Commit ae5a21e

Browse files
remove debugging changes to the eval script. fixes issue #26
1 parent d842e4a commit ae5a21e

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

vod/evaluation/kitti_official_evaluate.py

+3-3
Original file line numberDiff line numberDiff line change
@@ -411,15 +411,15 @@ def calculate_iou_partly(gt_annotations, dt_annotations, metric, num_parts=50):
411411
gt_annotations_part = gt_annotations[example_idx:example_idx + num_part]
412412
dt_annotations_part = dt_annotations[example_idx:example_idx + num_part]
413413
if metric == 0:
414-
gt_boxes = np.concatenate([(a["bbox"] + 0.01) for a in gt_annotations_part], 0)
414+
gt_boxes = np.concatenate([(a["bbox"]) for a in gt_annotations_part], 0)
415415
dt_boxes = np.concatenate([a["bbox"] for a in dt_annotations_part], 0)
416416
overlap_part = image_box_overlap(gt_boxes, dt_boxes)
417417
elif metric == 1:
418418
loc = np.concatenate(
419419
[a["location"][:, [0, 2]] for a in gt_annotations_part], 0)
420420
dims = np.concatenate(
421421
[a["dimensions"][:, [0, 2]] for a in gt_annotations_part], 0)
422-
rots = np.concatenate([(a["rotation_y"] + 0.01) for a in gt_annotations_part], 0)
422+
rots = np.concatenate([(a["rotation_y"]) for a in gt_annotations_part], 0)
423423
gt_boxes = np.concatenate(
424424
[loc, dims, rots[..., np.newaxis]], axis=1)
425425
loc = np.concatenate(
@@ -434,7 +434,7 @@ def calculate_iou_partly(gt_annotations, dt_annotations, metric, num_parts=50):
434434
elif metric == 2:
435435
loc = np.concatenate([a["location"] for a in gt_annotations_part], 0)
436436
dims = np.concatenate([a["dimensions"] for a in gt_annotations_part], 0)
437-
rots = np.concatenate([(a["rotation_y"] + 0.01) for a in gt_annotations_part], 0)
437+
rots = np.concatenate([(a["rotation_y"]) for a in gt_annotations_part], 0)
438438
gt_boxes = np.concatenate(
439439
[loc, dims, rots[..., np.newaxis]], axis=1)
440440
loc = np.concatenate([a["location"] for a in dt_annotations_part], 0)

0 commit comments

Comments
 (0)