@@ -411,15 +411,15 @@ def calculate_iou_partly(gt_annotations, dt_annotations, metric, num_parts=50):
411
411
gt_annotations_part = gt_annotations [example_idx :example_idx + num_part ]
412
412
dt_annotations_part = dt_annotations [example_idx :example_idx + num_part ]
413
413
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 )
415
415
dt_boxes = np .concatenate ([a ["bbox" ] for a in dt_annotations_part ], 0 )
416
416
overlap_part = image_box_overlap (gt_boxes , dt_boxes )
417
417
elif metric == 1 :
418
418
loc = np .concatenate (
419
419
[a ["location" ][:, [0 , 2 ]] for a in gt_annotations_part ], 0 )
420
420
dims = np .concatenate (
421
421
[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 )
423
423
gt_boxes = np .concatenate (
424
424
[loc , dims , rots [..., np .newaxis ]], axis = 1 )
425
425
loc = np .concatenate (
@@ -434,7 +434,7 @@ def calculate_iou_partly(gt_annotations, dt_annotations, metric, num_parts=50):
434
434
elif metric == 2 :
435
435
loc = np .concatenate ([a ["location" ] for a in gt_annotations_part ], 0 )
436
436
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 )
438
438
gt_boxes = np .concatenate (
439
439
[loc , dims , rots [..., np .newaxis ]], axis = 1 )
440
440
loc = np .concatenate ([a ["location" ] for a in dt_annotations_part ], 0 )
0 commit comments