omaralshareef/football-player-classifier

Football Player Classifier

Fine-tuned ResNet-50 on aaronqg/golden-foot-football-players.

Given a photo of a footballer, the model predicts which Golden Foot winner or nominee it is.

Classes

Alessandro Del Piero, Andrés Iniesta, Andriy Shevchenko, Cristiano Ronaldo, Didier Drogba, Diego Maradona, Edinson Cavani, Francesco Totti, Gianluigi Buffon, Iker Casillas, Lionel Messi, Luka Modrić, Mohamed Salah, Pavel Nedvěd, Pelé, Ryan Giggs, Roberto Baggio, Roberto Carlos, Ronaldinho, Ronaldo Nazário, Samuel Eto'o, Zlatan Ibrahimović.

Metrics

Split	Accuracy	Macro F1
Validation (best)	72.8%	72.7%
Test	74.1%	73.8%

Usage

from transformers import pipeline

clf = pipeline("image-classification", model="omaralshareef/football-player-classifier")
print(clf("player.jpg"))