NickSwardh / YoloDotNet

YoloDotNet - A C# .NET 8.0 project for Classification, Object Detection, OBB Detection, Segmentation and Pose Estimation in both images and videos.
GNU General Public License v3.0
158 stars 28 forks source link

IoU parameter for overlapping boxes #8

Closed sabihoshi closed 5 months ago

sabihoshi commented 5 months ago

There doesn't seem to be any IoU parameter from what I can see when I looked at the source code, I see that RemoveOverlappingBoxes uses a default value of 0.45f I think it would be useful if this was a parameter.

And maybe another IoU threshold parameter specifically if the overlapping boxes are only the same class, but I don't know if this is preferred. But I think at least one is very much needed.

NickSwardh commented 5 months ago

Hi Sabihoshi. Yes, the default for IoU is 0.45f. I've updated YoloDotNet to v1.5 with an additional parameter for controlling the IoU threshold if needed. Rock on! :)