This is a Pytorch implementation of Confused class cheking module for YOLOv5.
Can't see all of them to understand data features. It can be numerically checking how similar it is to which class and how much it is predicted with confidence score-based custom metric for model improvement.
- Calculate the variance of the confidence score within each image
- The smaller the variance value, the more confusing the class