Segformer B5 Finetuned Human Parsing

🚀 segformer-b5-finetuned-human-parsing

该模型是基于 nvidia/mit-b5 在未知数据集上进行微调的版本，可用于图像分割任务，在人体解析方面表现出色。

🚀 快速开始

此模型是 nvidia/mit-b5 在未知数据集上的微调版本。它在评估集上取得了以下结果：

• 损失值：0.2292
• 平均交并比（Mean Iou）：0.6258
• 平均准确率（Mean Accuracy）：0.7547
• 整体准确率（Overall Accuracy）：0.8256
• 背景准确率（Accuracy Background）：nan
• 帽子准确率（Accuracy Hat）：0.8561
• 头发准确率（Accuracy Hair）：0.8974
• 太阳镜准确率（Accuracy Sunglasses）：0.7540
• 上衣准确率（Accuracy Upper - clothes）：0.8553
• 裙子准确率（Accuracy Skirt）：0.7026
• 裤子准确率（Accuracy Pants）：0.8913
• 连衣裙准确率（Accuracy Dress）：0.7525
• 腰带准确率（Accuracy Belt）：0.4251
• 左脚鞋准确率（Accuracy Left - shoe）：0.6014
• 右脚鞋准确率（Accuracy Right - shoe）：0.6374
• 脸部准确率（Accuracy Face）：0.9094
• 左腿准确率（Accuracy Left - leg）：0.8452
• 右腿准确率（Accuracy Right - leg）：0.8343
• 左臂准确率（Accuracy Left - arm）：0.8506
• 右臂准确率（Accuracy Right - arm）：0.8287
• 包包准确率（Accuracy Bag）：0.8232
• 围巾准确率（Accuracy Scarf）：0.3662
• 背景交并比（Iou Background）：0.0
• 帽子交并比（Iou Hat）：0.7625
• 头发交并比（Iou Hair）：0.8171
• 太阳镜交并比（Iou Sunglasses）：0.6400
• 上衣交并比（Iou Upper - clothes）：0.7700
• 裙子交并比（Iou Skirt）：0.6211
• 裤子交并比（Iou Pants）：0.7788
• 连衣裙交并比（Iou Dress）：0.5512
• 腰带交并比（Iou Belt）：0.3564
• 左脚鞋交并比（Iou Left - shoe）：0.5032
• 右脚鞋交并比（Iou Right - shoe）：0.5381
• 脸部交并比（Iou Face）：0.8294
• 左腿交并比（Iou Left - leg）：0.7412
• 右腿交并比（Iou Right - leg）：0.7591
• 左臂交并比（Iou Left - arm）：0.7579
• 右臂交并比（Iou Right - arm）：0.7705
• 包包交并比（Iou Bag）：0.7729
• 围巾交并比（Iou Scarf）：0.2956

🔧 技术细节

训练超参数

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

• 学习率（learning_rate）：6e - 05
• 训练批次大小（train_batch_size）：2
• 评估批次大小（eval_batch_size）：2
• 随机种子（seed）：42
• 优化器（optimizer）：Adam，其中 betas=(0.9, 0.999)，epsilon = 1e - 08
• 学习率调度器类型（lr_scheduler_type）：线性
• 训练轮数（num_epochs）：15

训练结果

📄 许可证

该项目使用其他许可证。