Question 11: Consider the following code snippet that calculates the …

1import torch
2
3def calculate_iou(box1, box2):
4    # Calculate intersection
5    x1 = torch.max(box1[0], box2[0])
6    y1 = torch.max(box1[1], box2[1])
7    x2 = torch.min(box1[2], box2[2])
8    y2 = torch.min(box1[3], box2[3])
9    intersection = torch.clamp(x2 - x1, min=0) * torch.clamp(y2 - y1, min=0)
10
11    # Calculate union
12    area1 = (box1[2] - box1[0]) * (box1[3] - box1[1])
13    area2 = (box2[2] - box2[0]) * (box2[3] - box2[1])
14    union = area1 + area2 - intersection
15
16    # Compute IoU
17    iou = intersection / union if union > 0 else 0
18    return iou
19
20# Bounding boxes: [x1, y1, x2, y2]
21box1 = torch.tensor([0, 0, 2, 2])
22box2 = torch.tensor([1, 1, 3, 3])
23
24print(calculate_iou(box1, box2))