PyTorch-YOLOv3
PyTorch-YOLOv3 copied to clipboard
Confidence threshold per class
What I'm trying to do
I want to be able to have thersholds per class in the evaluate stage. Most significantly the confidence threshold, the nms threshold could also be useful.
What I've tried
I am still trying to figure out how to do it. Turning to the git to see if it was already done by someone.