Generalized IoU loss for Object Detection with Torchvision

A tutorial on how to use GIoU custom loss function in Torchvision object Detection

Maryam Bahrami
Towards Data Science
2 min readSep 5, 2021

--

Improvement of the Object Detection after using GIoU loss function (source)

In the object detection task, the most common evaluation metric is IoU, which motivates us to minimize such metrics in training and see if we make some progress or not. Hamid Rezatofighi and his colleagues showed that using the…

--

--