コード例 #1
0
ファイル: eval.py プロジェクト: tsbiosky/Cascade-point-rcnn 
def d3_box_overlap(boxes, qboxes, criterion=-1):
    rinc = rotate_iou_gpu_eval(boxes[:, [0, 2, 3, 5, 6]],
                               qboxes[:, [0, 2, 3, 5, 6]], 2)
    d3_box_overlap_kernel(boxes, qboxes, rinc, criterion)
    return rinc
コード例 #2
0
ファイル: eval.py プロジェクト: tsbiosky/Cascade-point-rcnn 
def bev_box_overlap(boxes, qboxes, criterion=-1):
    riou = rotate_iou_gpu_eval(boxes, qboxes, criterion)
    return riou