You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Hi,
in torchmetrics.detection.IntersectionOverUnion cases where a bounding box for a "non-existing" class are predicted are ignored. They should be 0 for that class instead.
Here "non-existing" means that the class does generally exist, just not in that image/examle.
To Reproduce
Take the snippet from the documentation and change the prediction:
🐛 Bug
Hi,
in
torchmetrics.detection.IntersectionOverUnion
cases where a bounding box for a "non-existing" class are predicted are ignored. They should be 0 for that class instead.Here "non-existing" means that the class does generally exist, just not in that image/examle.
To Reproduce
Take the snippet from the documentation and change the prediction:
The output is:
{'iou': tensor(0.6898), 'iou/cl_4': tensor(0.6898), 'iou/cl_5': tensor(nan)}
Expected behavior
First of all
iou/cl_5
should be 0.0 instead ofnan
(#2778).However I think there should also be an item with
'iou/cl_6': 0.0
.Environment
The text was updated successfully, but these errors were encountered: