I am pre-training a neural network for classifying three classes of images.
My datasets, due to their size, are difficult to filter out and clean up and there are many outliers which are not part of any class.
I would like to filter misclassified images out from the test image set (which is a datastore) based on their score and also set a parameter that classifies only if the scores is high enough (let's say 0.9). If the condition is not met, these images are separated from the dataset into a new folder, for example.
Is there a specific example or a script that would use these [YPred, scores] outputs and correlate them to the images that are falsly predicted in the test image set?
Thanks a lot in advance!