文章

Day39 - Neck中C3/C2f后添加一层CBAM在VisDrone上的表现

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

  • Yolov5n-cbam

    Parameters: 2.758M

    ClassPrecisionRecallmAP50mAP50-95
    all0.4320.3190.3180.182
    pedestrian0.4390.330.3320.141
    people0.5050.2180.2620.0905
    bicycle0.2240.09810.07440.029
    car0.6270.7420.7470.512
    van0.4570.3660.3640.252
    truck0.4390.2770.2830.181
    tricycle0.3740.2330.2110.111
    awning-tricycle0.260.1280.1240.078
    bus0.5520.4260.4320.279
    motor0.4430.3730.3490.144
  • Yolov5s-cbam

    Parameters: 9.564M

    ClassPrecisionRecallmAP50mAP50-95
    all0.5010.3670.3780.225
    pedestrian0.5360.3620.4010.182
    people0.5540.2590.3150.119
    bicycle0.250.1590.1150.046
    car0.7290.760.7860.557
    van0.5190.4340.4390.307
    truck0.5140.3330.3540.227
    tricycle0.4150.2860.2670.142
    awning-tricycle0.3340.1580.1510.0939
    bus0.6490.5020.5360.392
    motor0.5040.4170.4180.182
  • Yolov5m-cbam

    Parameters: 26.035M

    ClassPrecisionRecallmAP50mAP50-95
    all0.5270.4050.4170.252
    pedestrian0.5580.3960.4380.205
    people0.5880.2910.3460.137
    bicycle0.2690.1830.150.0652
    car0.7610.7770.8090.586
    van0.5520.4650.4770.34
    truck0.5530.380.3920.253
    tricycle0.4360.3360.3110.173
    awning-tricycle0.3080.2030.1660.103
    bus0.7020.5540.6050.445
    motor0.5470.470.470.216
  • Yolov8n-cbam

    Parameters: 3.260M

    ClassPrecisionRecallmAP50mAP50-95
    all0.3880.2890.2810.157
    pedestrian0.3820.3120.2870.118
    people0.4730.1910.2320.0803
    bicycle0.1910.05590.0520.0183
    car0.5960.7240.7190.485
    van0.4020.340.3240.215
    truck0.3340.2490.2180.137
    tricycle0.3370.1870.1790.0929
    awning-tricycle0.2250.1330.09910.0596
    bus0.5140.3780.3870.243
    motor0.4290.3170.3110.124
  • Yolov8s-cbam

    Parameters: 11.577M

    ClassPrecisionRecallmAP50mAP50-95
    all0.5020.3780.3860.231
    pedestrian0.5390.3770.4090.187
    people0.5430.2770.320.122
    bicycle0.2640.180.1310.0556
    car0.7390.7640.7910.565
    van0.5290.4380.4490.315
    truck0.5040.3410.3570.235
    tricycle0.3860.2920.2620.142
    awning-tricycle0.3020.1690.1530.0945
    bus0.7050.5060.560.401
    motor0.5080.4390.4320.194
本文由作者按照 CC BY 4.0 进行授权