Skip to main content
4 of 5
deleted 9 characters in body; edited tags; edited title
nbro
  • 41.4k
  • 12
  • 114
  • 205

How to detect multiple playing cards of the same class with a neural network?

I want to train an AI to detect playing cards. For that reason, I bought many different decks, scanned and labeled them. Next up would be to create training data with an augmentation library. I found two examples of how other people did that:

  1. Image detection using YOLO algorithm and poker cards
  2. Playing card detection with YOLO

The problem is that I want my AI to be able to detect multiple cards of the same class in one picture. In the examples, they put a label on the top left corner of a card. This makes sense since it is a very good indicator of what class the card is in. Unfortunately, every card has two of these labels.

I think their solution returns "detected" if any instance of a given label is found. But many cards with the same label could be in the same picture, and I am not sure if I can detect the quantity of the class with this solution.

Do you think it is feasible to mark the whole card, prepare the data accordingly, and train an AI that detects the count as well?