Category: PyTorch

Class Accuracies for Imbalanced Data in Deep Learning Image Recognition

Class Accuracies for Imbalanced Data in Deep Learning Image Recognition

In this article, we will discuss how to get per-class accuracy in a highly imbalanced image/vision dataset. Deep learning algorithms suffer when the dataset is highly imbalanced. In image recognition, a deep neural network may predict 90% of one class correctly and only 20% of another class correctly. This is the result of the number […] ...