Function example_wbf_1_model

example.py:107–136

This example shows how to ensemble boxes from single model using WBF method :return:

(iou_thr=0.55, draw_image=True)

def example_wbf_1_model(iou_thr=0.55, draw_image=True):
    """
    This example shows how to ensemble boxes from single model using WBF method    
    :return: 
    """

    boxes_list = [
        [0.00, 0.51, 0.81, 0.91],
        [0.10, 0.31, 0.71, 0.61],
        [0.01, 0.32, 0.83, 0.93],
        [0.02, 0.53, 0.11, 0.94],
        [0.03, 0.24, 0.12, 0.35],
        [0.04, 0.56, 0.84, 0.92],
        [0.12, 0.33, 0.72, 0.64],
        [0.38, 0.66, 0.79, 0.95],
        [0.08, 0.49, 0.21, 0.89],
    ]
    scores_list = [0.9, 0.8, 0.2, 0.4, 0.7, 0.5, 0.8, 0.7, 0.3]
    labels_list = [0, 1, 0, 1, 1, 1, 1, 1, 0]

    if draw_image:
        show_boxes([boxes_list], [scores_list], [labels_list])

    boxes, scores, labels = weighted_boxes_fusion([boxes_list], [scores_list], [labels_list], weights=None, iou_thr=iou_thr, skip_box_thr=0.0)

    if draw_image:
        show_boxes([boxes], [scores], [labels.astype(np.int32)])

    print(len(boxes))
    print(boxes)


def example_nms_2_models(method, iou_thr=0.5, sigma=0.5, thresh=0.001, draw_image=True):