def print_ious()

in mapillary_vistas/helpers/print.py [0:0]

• 31 lines of code
• 5 McCabe index (conditional complexity)

def print_ious(labels, confusion_matrix, instance_information=None):
    max_length = get_max_length(labels)
    iious = calculate_iiou(labels, confusion_matrix, instance_information)

    print('')
    print('')
    print('')

    header = "{:>{ml}}".format("", ml=max_length+1)
    header += " {:^6} |".format("IoU")
    header += " {:^6} |".format("iIoU")
    print(header)

    iou_values = []
    iiou_values = []
    for label, iiou in zip(labels, iious):
        if not math.isnan(iiou[0]):
            iou_values += [iiou[0]]
        current_line = "{:>{ml}}:".format(label["name"], ml=max_length)
        current_line += " {:>6} |".format("{:.1%}".format(iiou[0]))
        if label['instances']:
            if not math.isnan(iiou[1]):
                iiou_values += [iiou[1]]
            current_line += " {:>6} |".format("{:.1%}".format(iiou[1]))

        print(current_line)

    print('')
    print("{:>{ml}}: {:>6} | {:>6} |".format(
        "Avg",
        "{:.1%}".format(np.average(iou_values)),
        "{:.1%}".format(np.average(iiou_values)),
        ml=max_length
    ))
    print('')
    print('')