改进YOLOv5s的公路隧道烟火检测方法

马庆禄,鲁佳萍,唐小垚,段学锋

Improved YOLOv5s flame and smoke detection method in road tunnels

Qing-lu MA,Jia-ping LU,Xiao-yao TANG,Xue-feng DUAN

表 1 原YOLOv5s与YOLOv5s-PRO参数量及计算量的对比

Tab.1 Comparison of parameter quantity and calculation quantity of original YOLOv5s and YOLOv5s-PRO structure

序号 M1 P1/104 Q1/ GLOPs M1 P2/104 Q2/ GLOPs (P1—P2)/ 104 (Q1—Q2)/ GLOPs 0 Focus 0.3520 0.3539 Conv 0.0640 0.0590 0.2880 0.2949 1 CBL 1.8560 0.4719 CBL 1.8560 0.4719 0.0000 0.0000 2 BottleneckCSP 1.9904 0.4981 BottleneckCSP-2 1.7792 0.4456 0.2112 0.0524 3 CBL 7.3984 0.4719 CBAM 0.0610 0.0156 −0.0610 −0.0156 4 BottleneckCSP 16.1152 1.0224 CBL 7.3984 0.4719 0.0000 0.0000 5 CBL 29.5424 0.4719 BottleneckCSP-2 10.7392 0.6816 5.3760 0.3408 6 BottleneckCSP 64.1792 1.0224 CBAM 0.2146 0.0137 −0.2146 −0.0137 7 SPP 65.6896 0.2621 CBL 29.5424 0.4719 0.0000 0.0000 8 BottleneckCSP 124.8768 0.4981 BottleneckCSP-2 60.8768 0.9699 3.3024 0.0524 9 CBL 118.0672 0.4719 CBAM 0.8290 0.0133 −0.8290 −0.0133 10 — — — CBL 118.0672 0.4719 0.0000 0.0000 11 — — — BottleneckCSP-2 111.7184 0.4456 13.1584 0.0524 12 — — — CBAM 3.2866 0.0131 −3.2866 −0.0131 13 — — — SPPF 65.6896 0.2621 0.0000 0.0000 总量 — 430.0672 5.5443 — 412.1224 4.8071 17.9448 0.7372