一种人体属性识别的网络结构

0、前言

人体属性识别,是一个典型的多标签分类场景。每个人体有多个标签,如年龄、性别、衣着颜色等,而每个属性又有多种类别,如年龄分儿童青年老人、性别分男女、颜色分红绿青蓝紫...

本文提供了一个网络结构来执行这种任务。

1、网络结构

如图所示,网络接受一个批量的图像,经过backbone(骨干网络,可以是当前已开源的各种网络,如ResNet、EfficientNet、RepVGG等)提取特征,然后将此特征送入后续多个heads,每个head代表一个属性(性别、年龄等)的分类器(使用线性层构建),最后每个head即可输出在该属性上各个类别的概率(如性别head输出[女性概率,男性概率]):

一种人体属性识别的网络结构_第1张图片

你可能感兴趣的:(DeepLearning,人体属性识别,网络结构设计,多标签分类)