Submitted by Waterfront_xD t3_ydc9n1 in MachineLearning
Fapaak t1_itrsfcg wrote
A common problem in object detection is unbalanced dataset, if you cut out squares from the image and say 100 squares are background and 5 are the symbol, the model may not learn to distinguish it.
Try balancing the dataset if thats your issue and try to use Focal Loss
MLFanatic1337 t1_itsgvwa wrote
Also multi-instance learning methods could be helpful, in this case to be more resilient to background data.
Viewing a single comment thread. View all comments