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
