Calculate IoU
You have been asked to calculate the Intersection over Union (IoU) metric between each of the three predicted bounding boxes (box_a,box_b,box_c) and the ground truth box bbox.
Which predicted box is the closest to the ground truth?
box_iou has been imported from torchvision.ops.
box_a, box_b, box_c, bbox has been already converted to torch.int type tensors with an additional batch dimension.
This exercise is part of the course
Deep Learning for Images with PyTorch
Hands-on interactive exercise
Turn theory into action with one of our interactive exercises
Start Exercise