utils/matching.py

import numpy as np

def get_overlapping_part(damage_mask, part_masks, part_labels):
    max_iou = 0
    matched_part = None
    for mask, label in zip(part_masks, part_labels):
        iou = np.sum(np.logical_and(damage_mask, mask)) / np.sum(np.logical_or(damage_mask, mask))
        if iou > max_iou:
            max_iou = iou
            matched_part = label
    return matched_part, max_iou

def estimate_severity(area, part_label):
    if area < 2000:
        return "minor"
    elif area < 8000:
        return "moderate"
    else:
        return "severe" if part_label in ["front_bumper", "hood", "rear_bumper"] else "moderate"