Edit model card

Returns card type given an image.

See https://www.kaggle.com/code/dima806/card-types-image-detection-vit for more details.

Classification report:

                   precision    recall  f1-score   support

     ace of clubs     0.8000    0.9474    0.8675        38
  ace of diamonds     0.6604    0.9211    0.7692        38
    ace of hearts     0.7727    0.8947    0.8293        38
    ace of spades     0.6129    1.0000    0.7600        38
   eight of clubs     0.6500    0.3421    0.4483        38
eight of diamonds     0.7500    0.5385    0.6269        39
  eight of hearts     0.5000    0.1842    0.2692        38
  eight of spades     0.7273    0.2105    0.3265        38
    five of clubs     0.8438    0.6923    0.7606        39
 five of diamonds     0.7750    0.8158    0.7949        38
   five of hearts     0.7949    0.8158    0.8052        38
   five of spades     0.7368    0.7368    0.7368        38
    four of clubs     0.7333    0.8684    0.7952        38
 four of diamonds     0.8571    0.6316    0.7273        38
   four of hearts     0.7368    0.7368    0.7368        38
   four of spades     0.9000    0.6923    0.7826        39
    jack of clubs     0.7037    0.5000    0.5846        38
 jack of diamonds     0.5806    0.4737    0.5217        38
   jack of hearts     0.8889    0.2105    0.3404        38
   jack of spades     0.4000    0.2051    0.2712        39
            joker     0.9487    0.9737    0.9610        38
    king of clubs     0.3721    0.8421    0.5161        38
 king of diamonds     0.4865    0.9474    0.6429        38
   king of hearts     0.5472    0.7436    0.6304        39
   king of spades     0.4203    0.7632    0.5421        38
    nine of clubs     0.5909    0.6842    0.6341        38
 nine of diamonds     0.8095    0.4474    0.5763        38
   nine of hearts     0.5455    0.6154    0.5783        39
   nine of spades     0.4615    0.7895    0.5825        38
   queen of clubs     0.2727    0.1538    0.1967        39
queen of diamonds     0.6250    0.1282    0.2128        39
  queen of hearts     0.6216    0.6053    0.6133        38
  queen of spades     0.7353    0.6579    0.6944        38
   seven of clubs     0.5333    0.6316    0.5783        38
seven of diamonds     0.3571    0.3947    0.3750        38
  seven of hearts     0.7143    0.7895    0.7500        38
  seven of spades     0.7742    0.6316    0.6957        38
     six of clubs     0.7368    0.7179    0.7273        39
  six of diamonds     0.4462    0.7632    0.5631        38
    six of hearts     0.8462    0.5789    0.6875        38
    six of spades     0.7879    0.6842    0.7324        38
     ten of clubs     0.8889    0.6316    0.7385        38
  ten of diamonds     0.6136    0.7105    0.6585        38
    ten of hearts     0.7021    0.8684    0.7765        38
    ten of spades     0.8529    0.7632    0.8056        38
   three of clubs     0.7561    0.7949    0.7750        39
three of diamonds     0.7419    0.6053    0.6667        38
  three of hearts     0.7273    0.8205    0.7711        39
  three of spades     0.6744    0.7632    0.7160        38
     two of clubs     0.7179    0.7368    0.7273        38
  two of diamonds     0.7667    0.6053    0.6765        38
    two of hearts     0.7647    0.6842    0.7222        38
    two of spades     0.7949    0.8158    0.8052        38

         accuracy                         0.6553      2025
        macro avg     0.6804    0.6559    0.6431      2025
     weighted avg     0.6802    0.6553    0.6427      2025
Downloads last month
15
Safetensors
Model size
85.8M params
Tensor type
F32
·
Inference API
Drag image file here or click to browse from your device
This model can be loaded on Inference API (serverless).