# Load model directly
from transformers import AutoImageProcessor, AutoModelForObjectDetection
processor = AutoImageProcessor.from_pretrained("Genereux-akotenou/yolos-headwear")
model = AutoModelForObjectDetection.from_pretrained("Genereux-akotenou/yolos-headwear")Quick Links
YOLO-Face-4-KYC
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
Supported Categories
The model detects the following classes:
CATEGORIES = [
'face',
'glasses',
'hat',
'mask',
'headband',
'head covering'
]
- Downloads last month
- 472
Model tree for Genereux-akotenou/yolos-headwear
Base model
hustvl/yolos-small
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("object-detection", model="Genereux-akotenou/yolos-headwear")