| tags: | |
| - image-classification | |
| - pytorch | |
| datasets: | |
| - nateraw/fairface | |
| A vision transformer finetuned to classify the age of a given person's face. | |
| ```python | |
| import requests | |
| from PIL import Image | |
| from io import BytesIO | |
| from transformers import ViTFeatureExtractor, ViTForImageClassification | |
| # Get example image from official fairface repo + read it in as an image | |
| r = requests.get('https://github.com/dchen236/FairFace/blob/master/detected_faces/race_Asian_face0.jpg?raw=true') | |
| im = Image.open(BytesIO(r.content)) | |
| # Init model, transforms | |
| model = ViTForImageClassification.from_pretrained('nateraw/vit-age-classifier') | |
| transforms = ViTFeatureExtractor.from_pretrained('nateraw/vit-age-classifier') | |
| # Transform our image and pass it through the model | |
| inputs = transforms(im, return_tensors='pt') | |
| output = model(**inputs) | |
| # Predicted Class probabilities | |
| proba = output.logits.softmax(1) | |
| # Predicted Classes | |
| preds = proba.argmax(1) | |
| ``` |