Hello, does KotlinDL have an NMS (Non-Max Supression) method or do we have to impl it ourselves?
Alexandre Brown
05/12/2022, 11:32 AM
CC @zaleslaw@Julia Beliaeva
z
zaleslaw
05/13/2022, 12:41 PM
Is it an equivalent of argmax method? When you obtain the max element from the sequence of numbers?
a
Alexandre Brown
05/15/2022, 2:38 PM
I think it's more than just an argmax since it requires to compute the IoU of each bounding boxes as well.
You also might want to filter using an object score threshold