文章

Day45 - Backbone中C3/C2f后添加一层SEAttention在VisDrone上的表现

yolov5n/s/m和yolov8n/s仅在backbone的C3/C2f模块后添加一层SE,在VisDrone上训练200epochs

  • Yolov5n-se

    Parameters: 2.666M

    ClassPrecisionRecallmAP50mAP50-95
    all0.4170.3240.3170.181
    pedestrian0.4350.3320.3360.141
    people0.4830.2370.2660.0945
    bicycle0.20.1030.07270.0267
    car0.6270.7460.750.515
    van0.420.3890.3720.249
    truck0.4030.2860.260.163
    tricycle0.3440.2280.2030.109
    awning-tricycle0.2510.1320.110.0708
    bus0.5560.4180.4450.296
    motor0.450.3720.3570.145
  • Yolov5s-se

    Parameters: 9.197M

    ClassPrecisionRecallmAP50mAP50-95
    all0.4980.3590.3770.223
    pedestrian0.5420.360.4010.179
    people0.560.2610.3120.117
    bicycle0.2480.1440.1150.0477
    car0.7350.7610.7890.558
    van0.5140.4150.4380.305
    truck0.5220.3270.3540.225
    tricycle0.4120.2790.2630.143
    awning-tricycle0.3050.1580.1550.0983
    bus0.6330.4580.5190.374
    motor0.5080.4280.4220.182
  • Yolov5m-se

    Parameters: 25.209M

    ClassPrecisionRecallmAP50mAP50-95
    all0.5270.4030.4160.252
    pedestrian0.5620.4020.4430.208
    people0.6030.2970.3530.138
    bicycle0.2770.1790.1450.0624
    car0.7560.780.8050.584
    van0.550.4420.4640.328
    truck0.5350.3630.4010.263
    tricycle0.430.3240.3020.17
    awning-tricycle0.3420.2120.1820.112
    bus0.6690.5620.5890.434
    motor0.5450.4690.4710.216
  • Yolov8n-se

    Parameters: 3.168M

    ClassPrecisionRecallmAP50mAP50-95
    all0.4310.3330.3280.191
    pedestrian0.4510.3520.3550.153
    people0.4930.2410.2790.1
    bicycle0.2090.10.07890.0297
    car0.6570.7440.7550.522
    van0.4790.3630.3760.258
    truck0.4190.2950.2890.185
    tricycle0.3390.2450.2090.115
    awning-tricycle0.2570.1620.1150.0736
    bus0.5630.4380.4670.32
    motor0.4450.3870.3590.15
  • Yolov8s-se

    Parameters: 11.210M

    ClassPrecisionRecallmAP50mAP50-95
    all0.50.3730.3850.229
    pedestrian0.5480.3650.4080.184
    people0.5750.2650.3220.124
    bicycle0.2640.1410.1220.049
    car0.7430.7580.790.565
    van0.5180.430.4360.307
    truck0.5220.3410.3590.235
    tricycle0.4130.3080.2790.153
    awning-tricycle0.2880.160.1450.0885
    bus0.6260.5140.5420.389
    motor0.5030.4520.4420.195
本文由作者按照 CC BY 4.0 进行授权