File size: 634 Bytes
f01c86d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
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"
|