Table2
不同模型对比实验结果
Model | Precision | Recall | mAP@.0.50 | mAP@.0.50∶0.95 | Params(M) | GFLOPs |
YOLOv3 | 0.948 | 0.838 | 0.926 | 0.567 | 61.50 | 154.6 |
YOLOv3-tiny | 0.926 | 0.844 | 0.935 | 0.507 | 8.67 | 12.9 |
YOLOv3-SPP | 0.969 | 0.897 | 0.957 | 0.566 | 62.60 | 155.4 |
YOLOv5n | 0.808 | 0.823 | 0.847 | 0.499 | 1.58 | 3.9 |
YOLOv5s | 0.923 | 0.878 | 0.918 | 0.523 | 7.06 | 16.3 |
YOLOv6n | 0.944 | 0.850 | 0.932 | 0.579 | 4.30 | 11.1 |
YOLOv6s | 0.947 | 0.890 | 0.931 | 0.587 | 17.19 | 44.1 |
YOLOv7 | 0.919 | 0.887 | 0.952 | 0.558 | 36.50 | 103.2 |
YOLOv7-tiny | 0.937 | 0.840 | 0.909 | 0.541 | 6.01 | 13.0 |
测试模型 | 0.960 | 0.844 | 0.901 | 0.537 | 4.17 | 6.9 |
Table1
消融实验结果
Model | Precision | Recall | mAP@0.5 | mAP@0.5∶0.95 | Params(M) | GFLOPs |
YOLOv7-tiny | 0.937 | 0.840 | 0.909 | 0.541 | 6.01 | 13.0 |
YOLOv7-tiny+A | 0.942 | 0.818 | 0.874 | 0.502 | 4.16 | 6.9 |
YOLOv7-tiny+B | 0.896 | 0.893 | 0.942 | 0.580 | 6.02 | 13.1 |
YOLOv7-tiny+C | 0.935 | 0.878 | 0.929 | 0.562 | 6.01 | 13.0 |
YOLOv7-tiny+A+B+C | 0.960 | 0.844 | 0.901 | 0.537 | 4.17 | 6.9 |
注:A为在特征提取网络中引入MobileNetV3 -Small网络;B为在特征融合网络中融合CBAM注意力机制;C为使用改进的Focal-CIOU损失函数。 |
主办单位:煤炭科学研究总院有限公司 中国煤炭学会学术期刊工作委员会