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.
Deze oefening maakt deel uit van de cursus
Deep Learning for Images with PyTorch
Praktische interactieve oefening
Zet theorie om in actie met een van onze interactieve oefeningen.
Begin met trainen