Detr Resnet 50 Finetuned 10k Cppe5

🚀 detr-resnet-50-finetuned-10k-cppe5

该模型是 facebook/detr-resnet-50 在 cppe-5 数据集上的微调版本。它在目标检测任务中表现出色，能够精准识别特定目标。

🚀 快速开始

本模型是 facebook/detr-resnet-50 在 cppe-5 数据集上的微调版本。它在评估集上取得了以下结果：

• 损失值（Loss）：0.9865
• 平均精度均值（Map）：0.3578
• Map 50：0.6781
• Map 75：0.3105
• Map 小目标：0.3578
• Map 中目标：-1.0
• Map 大目标：-1.0
• 召回率均值（Mar）1：0.365
• Mar 10：0.535
• Mar 100：0.5483
• Mar 小目标：0.5483
• Mar 中目标：-1.0
• Mar 大目标：-1.0
• Map 全覆盖：0.6584
• Mar 100 全覆盖：0.7772
• Map 防护面罩：0.3691
• Mar 100 防护面罩：0.6063
• Map 手套：0.2477
• Mar 100 手套：0.4266
• Map 护目镜：0.1766
• Mar 100 护目镜：0.4655
• Map 口罩：0.3371
• Mar 100 口罩：0.4661

🔧 技术细节

训练超参数

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

• 学习率（learning_rate）：5e-05
• 训练批次大小（train_batch_size）：8
• 评估批次大小（eval_batch_size）：8
• 随机种子（seed）：1337
• 优化器（optimizer）：Adam，β1=0.9，β2=0.999，ε=1e-08
• 学习率调度器类型（lr_scheduler_type）：线性
• 训练轮数（num_epochs）：100.0
• 混合精度训练（mixed_precision_training）：原生自动混合精度（Native AMP）

训练结果

| 训练损失 | 轮数 | 步数 | 验证损失 | Map | Map 50 | Map 75 | Map 小目标 | Map 中目标 | Map 大目标 | Mar 1 | Mar 10 | Mar 100 | Mar 小目标 | Mar 中目标 | Mar 大目标 | Map 全覆盖 | Mar 100 全覆盖 | Map 防护面罩 | Mar 100 防护面罩 | Map 手套 | Mar 100 手套 | Map 护目镜 | Mar 100 护目镜 | Map 口罩 | Mar 100 口罩 | | ---- | ---- | ---- | ---- | ---- | ---- | ---- | ---- | ---- | ---- | ---- | ---- | ---- | ---- | ---- | ---- | ---- | ---- | ---- | ---- | ---- | ---- | ---- | ---- | ---- | | 3.7198 | 1.0 | 107 | 3.1869 | 0.0036 | 0.0144 | 0.0006 | 0.0036 | -1.0 | -1.0 | 0.017 | 0.0426 | 0.0647 | 0.0647 | -1.0 | -1.0 | 0.0169 | 0.1994 | 0.0 | 0.0 | 0.0001 | 0.0177 | 0.0 | 0.0 | 0.0008 | 0.1062 | | 3.0393 | 2.0 | 214 | 2.8978 | 0.007 | 0.021 | 0.003 | 0.007 | -1.0 | -1.0 | 0.0192 | 0.0796 | 0.1238 | 0.1238 | -1.0 | -1.0 | 0.0333 | 0.5216 | 0.0 | 0.0 | 0.0001 | 0.0292 | 0.0 | 0.0 | 0.0017 | 0.0684 | | ... | ... | ... | ... | ... | ... | ... | ... | ... | ... | ... | ... | ... | ... | ... | ... | ... | ... | ... | ... | ... | ... | ... | ... | ... | | 0.7089 | 100.0 | 10700 | 0.9865 | 0.3578 | 0.6781 | 0.3105 | 0.3578 | -1.0 | -1.0 | 0.365 | 0.535 | 0.5483 | 0.5483 | -1.0 | -1.0 | 0.6584 | 0.7772 | 0.3691 | 0.6063 | 0.2477 | 0.4266 | 0.1766 | 0.4655 | 0.3371 | 0.4661 |

📚 详细文档

框架版本

• Transformers：4.40.0.dev0
• Pytorch：1.13.0+cu117
• Datasets：2.18.0
• Tokenizers：0.19.0

📄 许可证

本模型采用 Apache-2.0 许可证。