yolos-headwear / README.md
Genereux-akotenou's picture
Create README.md
7a60a0d verified
|
raw
history blame
1.15 kB
metadata
datasets:
  - detection-datasets/fashionpedia
language:
  - en
pipeline_tag: object-detection
tags:
  - YOLOS
  - Object detection
  - KYC
  - Face detection
base_model:
  - hustvl/yolos-small
library_name: transformers

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'
]