Segformer B0 Finetuned Human Parsing

由 matei-dorian 开发

基于SegFormer-B0架构微调的人体解析模型，用于图像中人体部位的分割识别

图像分割

Transformers

开源协议:其他 #人体部件分割 #精细语义分割 #服装识别

下载量 16

发布时间 : 5/2/2023

模型介绍

内容详情

替代品

模型简介

该模型是基于MIT-B0架构在人体解析任务上微调的版本，能够识别和分割图像中的不同人体部位（如头发、上衣、裤子等）

模型特点

轻量级架构

基于SegFormer-B0架构，在保持较好性能的同时具有较小的模型尺寸

多类别分割

能够识别和分割20+个人体部位和服饰类别

迁移学习

基于预训练的MIT-B0模型进行微调，适应特定的人体解析任务

模型能力

图像分割

人体部位识别

服饰分类

语义分割

使用案例

时尚与零售

虚拟试衣

识别用户穿着以便进行虚拟服装叠加

时尚分析

分析图像中的服饰搭配和流行趋势

人机交互

增强现实应用

精确识别人体部位以实现更自然的AR交互

许可证: 其他标签:

视觉
图像分割
训练生成模型索引:
名称: segformer-b0-finetuned-human-parsing 结果: []

segformer-b0-finetuned-human-parsing

该模型是基于nvidia/mit-b0在None数据集上微调的版本。在评估集上取得了以下结果:

损失: 1.9476
平均交并比: 0.0726
平均准确率: 0.1221
总体准确率: 0.3575
背景准确率: 无
帽子准确率: 0.0048
头发准确率: 0.4813
太阳镜准确率: 0.0
上衣准确率: 0.9405
裙子准确率: 0.0000
裤子准确率: 0.0631
连衣裙准确率: 0.1031
腰带准确率: 0.0
左鞋准确率: 0.0011
右鞋准确率: 0.0010
面部准确率: 0.4406
左腿准确率: 0.0291
右腿准确率: 0.0
左臂准确率: 0.0
右臂准确率: 0.0001
包准确率: 0.0114
围巾准确率: 0.0
背景交并比: 0.0
帽子交并比: 0.0043
头发交并比: 0.4221
太阳镜交并比: 0.0
上衣交并比: 0.3239
裙子交并比: 0.0000
裤子交并比: 0.0559
连衣裙交并比: 0.0728
腰带交并比: 0.0
左鞋交并比: 0.0011
右鞋交并比: 0.0009
面部交并比: 0.3872
左腿交并比: 0.0271
右腿交并比: 0.0
左臂交并比: 0.0
右臂交并比: 0.0001
包交并比: 0.0106
围巾交并比: 0.0

模型描述

需要更多信息

预期用途与限制

需要更多信息

训练与评估数据

需要更多信息

训练过程

训练超参数

训练过程中使用了以下超参数:

学习率: 6e-05
训练批次大小: 2
评估批次大小: 2
随机种子: 42
优化器: Adam，参数β=(0.9,0.999)，ε=1e-08
学习率调度器类型: 线性
训练轮数: 2

训练结果

训练损失	轮次	步数	验证损失	平均交并比	平均准确率	总体准确率	背景准确率	帽子准确率	头发准确率	太阳镜准确率	上衣准确率	裙子准确率	裤子准确率	连衣裙准确率	左鞋准确率	右鞋准确率	面部准确率	左腿准确率	右腿准确率	左臂准确率	右臂准确率	包准确率	帽子交并比	头发交并比	太阳镜交并比	上衣交并比	裙子交并比	裤子交并比	连衣裙交并比	左鞋交并比	右鞋交并比	面部交并比	左腿交并比	右腿交并比	左臂交并比	右臂交并比	包交并比
2.5768	0.4	20	2.7812	0.0726	0.1332	0.2876	无	0.0178	0.3204	0.0004	0.5548	0.0004	0.2555	0.2373	0.0103	0.0003	0.5637	0.0287	0.0302	0.0001	0.0008	0.2435	0.0166	0.2759	0.0001	0.2781	0.0004	0.1710	0.1295	0.0098	0.0003	0.3251	0.0260	0.0248	0.0001	0.0007	0.0491
2.2093	0.8	40	2.5166	0.0563	0.1052	0.3288	无	0.0	0.1994	0.0	0.9447	0.0015	0.0435	0.1164	0.0008	0.0000	0.4655	0.0007	0.0003	0.0	0.0	0.0153	0.0	0.1946	0.0	0.3037	0.0015	0.0417	0.0842	0.0008	0.0000	0.3726	0.0007	0.0003	0.0	0.0	0.0124
1.8804	1.2	60	2.0209	0.0632	0.1110	0.3374	无	0.0087	0.3724	0.0	0.9475	0.0014	0.0162	0.0528	0.0001	0.0008	0.4257	0.0561	0.0001	0.0	0.0	0.0055	0.0077	0.3472	0.0	0.3086	0.0014	0.0156	0.0403	0.0001	0.0008	0.3597	0.0515	0.0001	0.0	0.0	0.0052
1.8776	1.6	80	2.0016	0.0665	0.1154	0.3454	无	0.0056	0.4172	0.0	0.9412	0.0000	0.0490	0.0697	0.0002	0.0006	0.4349	0.0329	0.0000	0.0	0.0000	0.0100	0.0048	0.3791	0.0	0.3138	0.0000	0.0438	0.0542	0.0002	0.0006	0.3608	0.0304	0.0000	0.0	0.0000	0.0093
1.8471	2.0	100	1.9476	0.0726	0.1221	0.3575	无	0.0048	0.4813	0.0	0.9405	0.0000	0.0631	0.1031	0.0011	0.0010	0.4406	0.0291	0.0	0.0	0.0001	0.0114	0.0043	0.4221	0.0	0.3239	0.0000	0.0559	0.0728	0.0011	0.0009	0.3872	0.0271	0.0	0.0	0.0001	0.0106