detection-datasets/fashionpedia
Viewer • Updated • 46.8k • 756 • 83
How to use Genereux-akotenou/yolos-headwear with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("object-detection", model="Genereux-akotenou/yolos-headwear") # Load model directly
from transformers import AutoImageProcessor, AutoModelForObjectDetection
processor = AutoImageProcessor.from_pretrained("Genereux-akotenou/yolos-headwear")
model = AutoModelForObjectDetection.from_pretrained("Genereux-akotenou/yolos-headwear")A fine-tuned YOLO model for detecting facial features and personal headwear, including glasses, hats, and masks — optimized for identity verification and KYC (Know Your Customer) applications. This model is built upon the YOLOS architecture and trained on a filtered and relabeled subset of the Fashionpedia dataset, focusing exclusively on categories relevant to liveness verification and facial compliance checks. For more details of the implementation, check out the source code here
The model detects the following classes:
CATEGORIES = [
'face',
'glasses',
'hat',
'mask',
'headband',
'head covering'
]

Base model
hustvl/yolos-small