基于全局信息感知的轻量级螺纹钢表面缺陷检测算法

肖剑,杨小苑,何昕泽,陈林,胡欣

Lightweight rebar surface defect detection algorithm based on global information perception

Jian XIAO,Xiaoyuan YANG,Xinze HE,Lin CHEN,Xin HU

表 3 损失函数对比实验

Tab.3 Comparison experiment of loss functions

损失函数 P/% R/% mAP@0.5/% mAP@0.5-0.95/% FPS/(帧·s−1) CIoU 88.3 83.1 90.2 55.9 133 DIoU 86.4 84.9 90.2 56.2 136 GIoU 87.9 83.6 90.0 55.9 131 EIoU 81.5 85.8 89.9 56.0 73 WIoU 81.5 85.3 89.8 55.8 93 SIoU 87.4 83.7 90.2 56.5 101 Wise-IoU 88.2 85.4 90.7 55.7 125 Inner-IoU 87.4 87.1 91.2 55.5 128 Unified-IoU(linear) 89.1 89.2 91.7 56.5 135 Unified-IoU(cos) 89.0 89.1 91.6 56.5 135 Unified-IoU(fraction) 88.8 89.1 91.3 56.6 136