基于改进RT-DETR的复杂天气下铁路异物实时检测算法
|
|
牛宏侠,冯鼎超,侯涛
|
Real-time detection algorithm for railway foreign objects in complex weather conditions based on improved RT-DETR
|
|
Hongxia NIU,Dingchao FENG,Tao HOU
|
|
| 表 4 改进模型中各模块的消融实验结果对比 |
| Tab.4 Comparison of ablation experiment results for each module in improved model |
|
| 组别 | FCM | RMFM | PSConv | mAP@0.5/% | mAP@0.5:0.95/% | Parameters | FLOPs/109 | F1-score | FPS/（帧·s−1） | | 1 | — | — | — | 94.02(±0.1) | 76.05 | 19880748 | 57.0 | 9208 | 75 | | 2 | √ | — | — | 94.88(±0.5) | 78.04(±0.5) | 6541688 | 47.9 | 9259 | 97 | | 3 | — | √ | — | 95.11(±0.3) | 78.21(±0.1) | 19716908 | 55.0 | 9261 | 77 | | 4 | — | — | √ | 95.09(±0.2) | 77.70(±0.1) | 19726224 | 58.2 | 9269 | 72 | | 5 | √ | √ | — | 95.21 | 79.01(±0.2) | 7002488 | 53.9 | 9357 | 92 | | 6 | √ | — | √ | 95.43(±0.1) | 79.11 | 7189550 | 56.7 | 9301 | 91 | | 7 | — | √ | √ | 95.18(±0.1) | 78.26 | 19887584 | 57.7 | 9308 | 71 | | 8 | √ | √ | √ | 95.67(±0.19) | 79.45(±0.3) | 7264632 | 56.8 | 9403 | 88 |
|
|
|