Panels Detection Rtdetr
P
Panels Detection Rtdetr
由 cems-official 开发
该模型是基于PekingU/rtdetr_r101vd_coco_o365微调的目标检测模型,专门用于检测船舶控制面板上的各种设备。
下载量 163
发布时间 : 11/19/2024
模型简介
该模型是一个基于RT-DETR架构的目标检测模型,经过微调后专门用于识别船舶控制面板上的各类设备,如雷达、船舶管理系统、目视观测设备等。
模型特点
多类别检测
能够检测船舶控制面板上的多种设备类型,包括雷达、船舶管理系统、目视观测设备等。
不同尺寸目标检测
能够有效检测不同尺寸的目标,包括小型、中型和大型设备。
高精度检测
在多个类别上达到较高的mAP值,特别是对船舶管理系统和目视观测设备的检测效果较好。
模型能力
船舶控制面板设备检测
多类别目标识别
不同尺寸目标检测
使用案例
船舶自动化
船舶控制面板监控
自动识别和监控船舶控制面板上的各种设备状态
可达到0.56的mAP值
航海设备管理
用于船舶设备的自动化管理和维护
对船舶管理系统的检测mAP可达0.76
航海安全
关键设备监控
监控船舶关键设备如雷达、推进器控制等的状态
雷达检测mAP可达0.58-0.70
🚀 面板检测RTDetr模型
本项目的panels_detection_rtdetr
模型是基于目标检测任务的微调模型,它在基础模型上进行优化,以适应特定场景的面板检测需求,能输出丰富的检测指标,为相关领域的应用提供有力支持。
🚀 快速开始
本模型是 PekingU/rtdetr_r101vd_coco_o365 在 None 数据集上的微调版本。它在评估集上取得了以下结果:
- 损失值(Loss):9.5718
- 平均精度均值(Map):0.5617
- 50% 交并比下的平均精度均值(Map 50):0.6631
- 75% 交并比下的平均精度均值(Map 75):0.6137
- 小目标的平均精度均值(Map Small):-1.0
- 中等目标的平均精度均值(Map Medium):0.3451
- 大目标的平均精度均值(Map Large):0.5935
- 1 个检测框下的平均召回率(Mar 1):0.6546
- 10 个检测框下的平均召回率(Mar 10):0.7877
- 100 个检测框下的平均召回率(Mar 100):0.8058
- 小目标在 100 个检测框下的平均召回率(Mar Small):-1.0
- 中等目标在 100 个检测框下的平均召回率(Mar Medium):0.5802
- 大目标在 100 个检测框下的平均召回率(Mar Large):0.8672
- 小雷达目标的平均精度均值(Map Radar (small)):0.3509
- 小雷达目标在 100 个检测框下的平均召回率(Mar 100 Radar (small)):0.8077
- 小型船舶管理系统目标的平均精度均值(Map Ship management system (small)):0.6748
- 小型船舶管理系统目标在 100 个检测框下的平均召回率(Mar 100 Ship management system (small)):0.8933
- 大型雷达目标的平均精度均值(Map Radar (large)):0.5846
- 大型雷达目标在 100 个检测框下的平均召回率(Mar 100 Radar (large)):0.8624
- 大型船舶管理系统目标的平均精度均值(Map Ship management system (large)):0.7577
- 大型船舶管理系统目标在 100 个检测框下的平均召回率(Mar 100 Ship management system (large)):0.9341
- 船舶管理系统顶部目标的平均精度均值(Map Ship management system (top)):0.789
- 船舶管理系统顶部目标在 100 个检测框下的平均召回率(Mar 100 Ship management system (top)):0.8356
- 大型电子海图显示与信息系统目标的平均精度均值(Map Ecdis (large)):0.3281
- 大型电子海图显示与信息系统目标在 100 个检测框下的平均召回率(Mar 100 Ecdis (large)):0.7652
- 小型视觉观察目标的平均精度均值(Map Visual observation (small)):0.585
- 小型视觉观察目标在 100 个检测框下的平均召回率(Mar 100 Visual observation (small)):0.902
- 小型电子海图显示与信息系统目标的平均精度均值(Map Ecdis (small)):0.7635
- 小型电子海图显示与信息系统目标在 100 个检测框下的平均召回率(Mar 100 Ecdis (small)):0.8967
- 船舶管理系统桌面目标的平均精度均值(Map Ship management system (table top)):0.6306
- 船舶管理系统桌面目标在 100 个检测框下的平均召回率(Mar 100 Ship management system (table top)):0.7882
- 推进器控制目标的平均精度均值(Map Thruster control):0.4949
- 推进器控制目标在 100 个检测框下的平均召回率(Mar 100 Thruster control):0.7447
- 左侧视觉观察目标的平均精度均值(Map Visual observation (left)):0.6062
- 左侧视觉观察目标在 100 个检测框下的平均召回率(Mar 100 Visual observation (left)):0.8395
- 中间视觉观察目标的平均精度均值(Map Visual observation (mid)):0.7946
- 中间视觉观察目标在 100 个检测框下的平均召回率(Mar 100 Visual observation (mid)):0.8901
- 右侧视觉观察目标的平均精度均值(Map Visual observation (right)):0.7446
- 右侧视觉观察目标在 100 个检测框下的平均召回率(Mar 100 Visual observation (right)):0.8966
- 船首推进器目标的平均精度均值(Map Bow thruster):0.2392
- 船首推进器目标在 100 个检测框下的平均召回率(Mar 100 Bow thruster):0.5167
- 主机传令钟目标的平均精度均值(Map Me telegraph):0.0825
- 主机传令钟目标在 100 个检测框下的平均召回率(Mar 100 Me telegraph):0.5143
📚 详细文档
训练过程
训练超参数
训练过程中使用了以下超参数:
- 学习率(learning_rate):5e - 05
- 训练批次大小(train_batch_size):4
- 评估批次大小(eval_batch_size):8
- 随机种子(seed):42
- 优化器(optimizer):使用 adamw_torch,其中 betas = (0.9, 0.999),epsilon = 1e - 08,无额外优化器参数
- 学习率调度器类型(lr_scheduler_type):余弦退火
- 训练轮数(num_epochs):7
训练结果
训练损失 | 轮数 | 步数 | 验证损失 | 平均精度均值 | 50% 交并比下的平均精度均值 | 75% 交并比下的平均精度均值 | 小目标的平均精度均值 | 中等目标的平均精度均值 | 大目标的平均精度均值 | 1 个检测框下的平均召回率 | 10 个检测框下的平均召回率 | 100 个检测框下的平均召回率 | 小目标在 100 个检测框下的平均召回率 | 中等目标在 100 个检测框下的平均召回率 | 大目标在 100 个检测框下的平均召回率 | 小雷达目标的平均精度均值 | 小雷达目标在 100 个检测框下的平均召回率 | 小型船舶管理系统目标的平均精度均值 | 小型船舶管理系统目标在 100 个检测框下的平均召回率 | 大型雷达目标的平均精度均值 | 大型雷达目标在 100 个检测框下的平均召回率 | 大型船舶管理系统目标的平均精度均值 | 大型船舶管理系统目标在 100 个检测框下的平均召回率 | 船舶管理系统顶部目标的平均精度均值 | 船舶管理系统顶部目标在 100 个检测框下的平均召回率 | 大型电子海图显示与信息系统目标的平均精度均值 | 大型电子海图显示与信息系统目标在 100 个检测框下的平均召回率 | 小型视觉观察目标的平均精度均值 | 小型视觉观察目标在 100 个检测框下的平均召回率 | 小型电子海图显示与信息系统目标的平均精度均值 | 小型电子海图显示与信息系统目标在 100 个检测框下的平均召回率 | 船舶管理系统桌面目标的平均精度均值 | 船舶管理系统桌面目标在 100 个检测框下的平均召回率 | 推进器控制目标的平均精度均值 | 推进器控制目标在 100 个检测框下的平均召回率 | 左侧视觉观察目标的平均精度均值 | 左侧视觉观察目标在 100 个检测框下的平均召回率 | 中间视觉观察目标的平均精度均值 | 中间视觉观察目标在 100 个检测框下的平均召回率 | 右侧视觉观察目标的平均精度均值 | 右侧视觉观察目标在 100 个检测框下的平均召回率 | 船首推进器目标的平均精度均值 | 船首推进器目标在 100 个检测框下的平均召回率 | 主机传令钟目标的平均精度均值 | 主机传令钟目标在 100 个检测框下的平均召回率 |
---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
14.2599 | 1.0 | 699 | 9.6242 | 0.4769 | 0.5404 | 0.5144 | -1.0 | 0.2274 | 0.5416 | 0.5866 | 0.755 | 0.7709 | -1.0 | 0.4884 | 0.8359 | 0.7408 | 0.92 | 0.672 | 0.8827 | 0.7054 | 0.9504 | 0.8329 | 0.926 | 0.7965 | 0.8692 | 0.3419 | 0.9571 | 0.2734 | 0.8627 | 0.1207 | 0.6933 | 0.4841 | 0.7059 | 0.3541 | 0.6947 | 0.5303 | 0.8961 | 0.8393 | 0.9342 | 0.2629 | 0.8466 | 0.1988 | 0.3583 | 0.0011 | 0.0667 |
8.9356 | 2.0 | 1398 | 9.1941 | 0.5527 | 0.6652 | 0.6044 | -1.0 | 0.3212 | 0.574 | 0.6512 | 0.7882 | 0.8015 | -1.0 | 0.6608 | 0.8085 | 0.6989 | 0.8862 | 0.5273 | 0.8053 | 0.7683 | 0.9145 | 0.7209 | 0.9073 | 0.7995 | 0.8644 | 0.4929 | 0.833 | 0.4034 | 0.8392 | 0.5519 | 0.8333 | 0.6453 | 0.8618 | 0.4221 | 0.6447 | 0.5734 | 0.8474 | 0.8714 | 0.8973 | 0.412 | 0.8448 | 0.3154 | 0.5333 | 0.0874 | 0.5095 |
8.1388 | 3.0 | 2097 | 9.7524 | 0.535 | 0.6013 | 0.5854 | -1.0 | 0.2545 | 0.574 | 0.6219 | 0.7425 | 0.7612 | -1.0 | 0.538 | 0.8183 | 0.6358 | 0.8292 | 0.5844 | 0.8013 | 0.6721 | 0.8368 | 0.7422 | 0.8829 | 0.7144 | 0.8096 | 0.4904 | 0.8562 | 0.7623 | 0.9078 | 0.5667 | 0.89 | 0.6409 | 0.7824 | 0.1853 | 0.5763 | 0.5453 | 0.7789 | 0.8362 | 0.9 | 0.5862 | 0.9207 | 0.0384 | 0.3833 | 0.0248 | 0.2619 |
7.5951 | 4.0 | 2796 | 9.3983 | 0.5991 | 0.7001 | 0.6587 | -1.0 | 0.3745 | 0.6167 | 0.6957 | 0.8036 | 0.8188 | -1.0 | 0.6611 | 0.8746 | 0.603 | 0.8538 | 0.626 | 0.88 | 0.6211 | 0.8496 | 0.8218 | 0.9382 | 0.8062 | 0.8433 | 0.3917 | 0.8804 | 0.6202 | 0.851 | 0.8307 | 0.9433 | 0.555 | 0.8147 | 0.5143 | 0.8 | 0.6609 | 0.8579 | 0.887 | 0.9369 | 0.7174 | 0.8759 | 0.2732 | 0.5333 | 0.0579 | 0.4238 |
7.1786 | 5.0 | 3495 | 9.1194 | 0.6117 | 0.7144 | 0.6689 | -1.0 | 0.3458 | 0.6476 | 0.6904 | 0.8136 | 0.8324 | -1.0 | 0.6649 | 0.8777 | 0.5 | 0.8538 | 0.6723 | 0.8733 | 0.7272 | 0.8795 | 0.778 | 0.9398 | 0.7803 | 0.8385 | 0.3389 | 0.8509 | 0.6484 | 0.8804 | 0.7914 | 0.9433 | 0.7053 | 0.8059 | 0.6257 | 0.8447 | 0.5945 | 0.8658 | 0.8411 | 0.9009 | 0.7812 | 0.9397 | 0.2863 | 0.5792 | 0.1053 | 0.4905 |
7.1386 | 6.0 | 4194 | 9.9394 | 0.5353 | 0.634 | 0.5921 | -1.0 | 0.3062 | 0.5549 | 0.6429 | 0.7691 | 0.7874 | -1.0 | 0.5638 | 0.8364 | 0.3431 | 0.7631 | 0.6563 | 0.8813 | 0.5789 | 0.8393 | 0.6941 | 0.9236 | 0.721 | 0.7712 | 0.4061 | 0.8018 | 0.5685 | 0.8725 | 0.7656 | 0.91 | 0.5317 | 0.8 | 0.5194 | 0.7684 | 0.5191 | 0.8039 | 0.7994 | 0.8586 | 0.6714 | 0.8793 | 0.2223 | 0.4958 | 0.0333 | 0.4429 |
7.0912 | 7.0 | 4893 | 9.5718 | 0.5617 | 0.6631 | 0.6137 | -1.0 | 0.3451 | 0.5935 | 0.6546 | 0.7877 | 0.8058 | -1.0 | 0.5802 | 0.8672 | 0.3509 | 0.8077 | 0.6748 | 0.8933 | 0.5846 | 0.8624 | 0.7577 | 0.9341 | 0.789 | 0.8356 | 0.3281 | 0.7652 | 0.585 | 0.902 | 0.7635 | 0.8967 | 0.6306 | 0.7882 | 0.4949 | 0.7447 | 0.6062 | 0.8395 | 0.7946 | 0.8901 | 0.7446 | 0.8966 | 0.2392 | 0.5167 | 0.0825 | 0.5143 |
框架版本
- Transformers 4.46.0
- Pytorch 2.5.0+cu121
- Datasets 3.0.2
- Tokenizers 0.20.1
📄 许可证
本项目采用 Apache - 2.0 许可证。
Table Transformer Detection
MIT
基于DETR架构的表格检测模型,专门用于从非结构化文档中提取表格
目标检测
Transformers

T
microsoft
2.6M
349
Grounding Dino Base
Apache-2.0
Grounding DINO是一个开放集目标检测模型,通过结合DINO检测器与文本编码器实现零样本目标检测能力。
目标检测
Transformers

G
IDEA-Research
1.1M
87
Grounding Dino Tiny
Apache-2.0
Grounding DINO是一个结合DINO检测器与接地预训练的开放集目标检测模型,能够实现零样本目标检测。
目标检测
Transformers

G
IDEA-Research
771.67k
74
Detr Resnet 50
Apache-2.0
DETR是一个基于Transformer架构的端到端目标检测模型,使用ResNet-50作为骨干网络,在COCO数据集上训练。
目标检测
Transformers

D
facebook
505.27k
857
Detr Resnet 101
Apache-2.0
DETR是一个使用Transformer架构的端到端目标检测模型,采用ResNet-101作为骨干网络,在COCO数据集上训练。
目标检测
Transformers

D
facebook
262.94k
119
Detr Doc Table Detection
Apache-2.0
基于DETR架构的文档表格检测模型,用于检测文档中的有边框和无边框表格
目标检测
Transformers

D
TahaDouaji
233.45k
59
Yolos Small
Apache-2.0
基于视觉Transformer(ViT)的目标检测模型,使用DETR损失函数训练,在COCO数据集上表现优异。
目标检测
Transformers

Y
hustvl
154.46k
63
Yolos Tiny
Apache-2.0
基于COCO 2017目标检测数据集微调的YOLOS模型,使用视觉Transformer架构实现高效目标检测。
目标检测
Transformers

Y
hustvl
144.58k
266
Rtdetr R50vd Coco O365
Apache-2.0
RT-DETR是首个实时端到端目标检测器,通过高效混合编码器和不确定性最小化查询选择机制,在COCO数据集上达到53.1% AP,108 FPS的性能。
目标检测
Transformers 英语

R
PekingU
111.17k
11
Rtdetr R101vd Coco O365
Apache-2.0
首个实时端到端目标检测器,基于Transformer架构,消除非极大值抑制需求,在速度与精度上超越YOLO系列
目标检测
Transformers 英语

R
PekingU
106.81k
7
精选推荐AI模型
Llama 3 Typhoon V1.5x 8b Instruct
专为泰语设计的80亿参数指令模型,性能媲美GPT-3.5-turbo,优化了应用场景、检索增强生成、受限生成和推理任务
大型语言模型
Transformers 支持多种语言

L
scb10x
3,269
16
Cadet Tiny
Openrail
Cadet-Tiny是一个基于SODA数据集训练的超小型对话模型,专为边缘设备推理设计,体积仅为Cosmo-3B模型的2%左右。
对话系统
Transformers 英语

C
ToddGoldfarb
2,691
6
Roberta Base Chinese Extractive Qa
基于RoBERTa架构的中文抽取式问答模型,适用于从给定文本中提取答案的任务。
问答系统 中文
R
uer
2,694
98