Get startedGet started for free

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

View Course

Hands-on interactive exercise

Turn theory into action with one of our interactive exercises

Start Exercise