Returns celebrity name given a facial image with about 93% accuracy.
See https://www.kaggle.com/code/dima806/celebs-face-image-detection-vit for more details.
Classification report:
precision recall f1-score support
Adriana Lima 0.9462 0.9362 0.9412 94
Alex Lawther 0.9490 0.9789 0.9637 95
Alexandra Daddario 0.9485 0.9684 0.9583 95
Alvaro Morte 0.9794 1.0000 0.9896 95
Alycia Dabnem Carey 0.9620 0.8000 0.8736 95
Amanda Crew 0.9286 0.9579 0.9430 95
Amber Heard 0.8652 0.8105 0.8370 95
Andy Samberg 0.9785 0.9681 0.9733 94
Anne Hathaway 0.9109 0.9684 0.9388 95
Anthony Mackie 1.0000 1.0000 1.0000 95
Avril Lavigne 0.9135 1.0000 0.9548 95
Barack Obama 1.0000 1.0000 1.0000 95
Barbara Palvin 0.9175 0.9368 0.9271 95
Ben Affleck 0.9474 0.9474 0.9474 95
Bill Gates 1.0000 1.0000 1.0000 95
Bobby Morley 0.9400 0.9895 0.9641 95
Brenton Thwaites 0.9474 0.9574 0.9524 94
Brian J. Smith 0.8559 1.0000 0.9223 95
Brie Larson 0.8558 0.9368 0.8945 95
Camila Mendes 0.9495 0.9895 0.9691 95
Chris Evans 0.9247 0.9053 0.9149 95
Chris Hemsworth 0.9565 0.9263 0.9412 95
Chris Pratt 0.9691 0.9895 0.9792 95
Christian Bale 0.9783 0.9574 0.9677 94
Cristiano Ronaldo 1.0000 1.0000 1.0000 94
Danielle Panabaker 0.9859 0.7368 0.8434 95
Dominic Purcell 0.9792 0.9895 0.9843 95
Dwayne Johnson 0.9895 1.0000 0.9947 94
Eliza Taylor 0.9750 0.8211 0.8914 95
Elizabeth Lail 0.9670 0.9263 0.9462 95
Elizabeth Olsen 0.8411 0.9474 0.8911 95
Ellen Page 0.8687 0.9053 0.8866 95
Elon Musk 0.9583 0.9684 0.9634 95
Emilia Clarke 0.9206 0.6105 0.7342 95
Emma Stone 0.9500 0.8000 0.8686 95
Emma Watson 0.9615 0.5263 0.6803 95
Gal Gadot 0.9296 0.6947 0.7952 95
Grant Gustin 0.9468 0.9368 0.9418 95
Gwyneth Paltrow 0.8796 1.0000 0.9360 95
Henry Cavil 0.9487 0.7789 0.8555 95
Hugh Jackman 0.9570 0.9368 0.9468 95
Inbar Lavi 0.9570 0.9368 0.9468 95
Irina Shayk 0.9592 0.9895 0.9741 95
Jake Mcdorman 1.0000 0.9789 0.9894 95
Jason Momoa 0.9894 0.9789 0.9841 95
Jeff Bezos 0.9896 1.0000 0.9948 95
Jennifer Lawrence 0.8876 0.8404 0.8634 94
Jeremy Renner 0.9691 0.9895 0.9792 95
Jessica Barden 0.8624 1.0000 0.9261 94
Jimmy Fallon 0.9792 0.9895 0.9843 95
Johnny Depp 0.9140 0.8947 0.9043 95
Josh Radnor 0.9792 0.9895 0.9843 95
Katharine Mcphee 0.9333 0.8842 0.9081 95
Katherine Langford 0.7851 1.0000 0.8796 95
Keanu Reeves 0.9785 0.9579 0.9681 95
Kiernen Shipka 0.6078 0.9789 0.7500 95
Krysten Ritter 0.9118 0.9894 0.9490 94
Leonardo Dicaprio 0.9588 0.9789 0.9688 95
Lili Reinhart 0.8144 0.8404 0.8272 94
Lindsey Morgan 0.8571 0.9474 0.9000 95
Lionel Messi 0.9890 0.9474 0.9677 95
Logan Lerman 0.9583 0.9684 0.9634 95
Madelaine Petsch 0.9072 0.9362 0.9215 94
Maisie Williams 0.8713 0.9362 0.9026 94
Margot Robbie 0.7634 0.7474 0.7553 95
Maria Pedraza 0.9310 0.8617 0.8950 94
Marie Avgeropoulos 0.9118 0.9789 0.9442 95
Mark Ruffalo 1.0000 0.8632 0.9266 95
Mark Zuckerberg 0.9896 1.0000 0.9948 95
Megan Fox 1.0000 0.9362 0.9670 94
Melissa Fumero 0.9400 0.9895 0.9641 95
Miley Cyrus 1.0000 0.7053 0.8272 95
Millie Bobby Brown 0.9192 0.9579 0.9381 95
Morena Baccarin 0.9789 0.9789 0.9789 95
Morgan Freeman 1.0000 1.0000 1.0000 94
Nadia Hilker 0.9892 0.9787 0.9840 94
Natalie Dormer 0.7417 0.9368 0.8279 95
Natalie Portman 0.8804 0.8526 0.8663 95
Neil Patrick Harris 1.0000 0.9789 0.9894 95
Pedro Alonso 0.9579 0.9579 0.9579 95
Penn Badgley 0.9583 0.9787 0.9684 94
Rami Malek 0.9792 0.9895 0.9843 95
Rebecca Ferguson 0.8304 0.9789 0.8986 95
Richard Harmon 0.9381 0.9579 0.9479 95
Rihanna 0.9485 0.9787 0.9634 94
Robert De Niro 0.8687 0.9053 0.8866 95
Robert Downey Jr 0.9765 0.8830 0.9274 94
Sarah Wayne Callies 0.8476 0.9368 0.8900 95
Scarlett Johansson 0.9302 0.4211 0.5797 95
Selena Gomez 0.9359 0.7684 0.8439 95
Shakira Isabel Mebarak 0.9368 0.9368 0.9368 95
Sophie Turner 0.8969 0.9158 0.9062 95
Stephen Amell 0.9500 1.0000 0.9744 95
Taylor Swift 0.9300 0.9789 0.9538 95
Tom Cruise 0.9688 0.9789 0.9738 95
Tom Ellis 0.9208 0.9894 0.9538 94
Tom Hardy 0.9765 0.8737 0.9222 95
Tom Hiddleston 0.9451 0.9053 0.9247 95
Tom Holland 0.9300 0.9789 0.9538 95
Tuppence Middleton 0.8304 0.9789 0.8986 95
Ursula Corbero 0.9278 0.9474 0.9375 95
Wentworth Miller 0.9694 1.0000 0.9845 95
Zac Efron 0.9192 0.9579 0.9381 95
Zendaya 0.8468 0.9895 0.9126 95
Zoe Saldana 1.0000 1.0000 1.0000 94
accuracy 0.9277 9954
macro avg 0.9324 0.9277 0.9260 9954
weighted avg 0.9324 0.9277 0.9259 9954
- Downloads last month
- 621
This model does not have enough activity to be deployed to Inference API (serverless) yet. Increase its social
visibility and check back later, or deploy to Inference Endpoints (dedicated)
instead.
Model tree for dima806/celebs_face_image_detection
Base model
google/vit-base-patch16-224-in21k