osanseviero's picture
Add tag for discoverability
b4f67fa
metadata
tags:
  - fastai
  - resnet
  - computer-vision
  - classification
  - image-classification
  - binary-classification
license:
  - cc0-1.0

Resnet34 Pokemon Card Classifier

Model Description

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.

Intended uses & limitation

This model is trained to identify real vs fake cards based on the backs of the cards, not the front.

How to use

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)

Training data

Dataset located here: link.