🔥 基于YOLOv8的人脸属性分类模型的构建

# 本项目采用的是yolov8s-cls模型

from ultralytics import YOLO

# Load a model
model = YOLO("yolov8s-cls.pt")  # load a pretrained model (recommended for training)

# Train the model
results = model.train(data="path/to/dataset", epochs=200, imgsz=640)

幼儿：1周岁-6周岁。
儿童：7周岁-14周岁。
少年：15周岁-19周岁。
青年：20周岁-39周岁。
中年：40周岁-59周岁。
老年：60周岁以上。

Infants: 1 year old to 6 years old.
Children: 7 years old - 14 years old.
Teenagers: 13 years old - 19 years old.
Youth: 20 years old - 39 years old.
Middle: 40 years old to 59 years old.
Elderly: Over 60 years old.

# 年龄分类字典
{0: 'Children', 1: 'Elderly', 2: 'Infants', 3: 'Middle', 4: 'Teenagers', 5: 'Youth'}

man
woman

# 性别分类字典
{0: 'man', 1: 'woman'}