This is a resnet34 model fine-tuned with fastai to classify real and fake Pokemon cards (dataset).
Here is a colab notebook that shows how the model was trained and pushed to the hub: link.
This model is trained to identify real vs fake cards based on the backs of the cards, not the front.
from huggingface_hub import from_pretrained_fastai # Pull model from hub learn = from_pretrained_fastai('hugginglearners/pokemon-card-checker') # Get prediction for this image pred_label, _, scores = learn.predict(img)
Dataset located here: link.
- Downloads last month