文章

Day15

参照HIC-YOLO把yolov8-cbam里的cbam模块放到最后, 跑一遍yolov8m-cbam、yolov8m-p234、yolov5m-cbam

yolov5m-p2的结果也跑出来了,跟上次的合并一下

epochs均为200,数据集均为VisDrone2019

ModelParametersGFLOPsTrain EpochsBest EpochsmAP@50:95mAP50
yolov8m-cbam49431668193.81681180.248420.41506
yolov8m2590264079.31741240.257430.42222
yolov8m-p225544952100.61771270.293080.47413
yolov5m2511145664.62001500.251890.41390
yolov5m-resnet42321966106.7200 0.238950.39860
yolov5m-p223862768NoneNoneNone0.282830.46070

可以发现多了个检测头对精度提升效果很大,足足提高了5个点,但是resnet并没有减少很多参数量,怀疑是代码有问题,需要进一步测试修改,接下来测试把P4检测头删了,看看参数量有啥变化,对精度提升效果如何

还有cbam的修改,按理说应该涨点,为啥会掉呢

很高兴人生第一个pr被merge了

本文由作者按照 CC BY 4.0 进行授权