torchmetrics icon indicating copy to clipboard operation
torchmetrics copied to clipboard

Add IoU/GIoU for boxes

Open twsl opened this issue 3 years ago • 4 comments

🚀 Feature

Motivation

IoU/GIoU are ametrics essential for the detection domain.

Pitch

Add metrics, for fast implementation these could be based on torchvisions functions.

Alternatives

Additional context

twsl avatar Dec 06 '21 08:12 twsl

I think we can use this code, right? https://github.com/PyTorchLightning/lightning-bolts/blob/0.5.0/pl_bolts/metrics/object_detection.py

Borda avatar Dec 21 '21 10:12 Borda

There is an even easier solution if we use torchvision. I can create a draft tomorrow

twsl avatar Dec 21 '21 15:12 twsl

There is an even easier solution if we use torchvision. I can create a draft tomorrow

I would try to keep own implementations if possible :rabbit:

Borda avatar Dec 21 '21 15:12 Borda

@Borda - I can take this issue :)

jscottcronin avatar Jan 14 '22 20:01 jscottcronin