admin管理员组

文章数量:1487745

YoloV8改进策略:复现HIC

摘要

HIC-YOLOv5主要贡献可以总结如下:

  • 额外的预测头专为小物体设计。它在更高分辨率的特征图中检测物体,这些特征图包含更多关于微小和小物体的信息。
  • 添加了一个内卷积块作为主干和颈部之间的桥梁,以增加特征图的通道信息。
  • 在主干的末端应用了CBAM,从而提取了更多重要的通道和空间信息,同时忽略了冗余的信息。

结果表明,HIC-YOLOv5在VisDrone-2019-DET数据集上的mAP@[.5:.95]提高了6.42%,mAP@0.5提高了9.38%。

我们参照HIC-YOLOv5,将这些改进用于YoloV8,效果如何呢?我们一起见证吧!

Yolov8官方结果

代码语言:javascript代码运行次数:0运行复制
YOLOv8l summary (fused): 268 layers, 43631280 parameters, 0 gradients, 165.0 GFLOPs
                 Class     Images  Instances      Box(P          R      mAP50  mAP50-95): 100%|██████████| 29/29 [
                   all        230       1412      0.922      0.957      0.986      0.737
                   c17        230        131      0.973      0.992      0.995      0.825
                    c5        230         68      0.945          1      0.995      0.836
            helicopter        230         43       0.96      0.907      0.951      0.607
                  c130        230         85      0.984          1      0.995      0.655
                   f16        230         57      0.955      0.965      0.985      0.669
                    b2        230          2      0.704          1      0.995      0.722
                 other        230         86      0.903      0.942      0.963      0.534
                   b52        230         70       0.96      0.971      0.978      0.831
                  kc10        230         62      0.999      0.984       0.99      0.847
               command        230         40       0.97          1      0.995      0.811
                   f15        230        123      0.891          1      0.992      0.701
                 kc135        230         91      0.971      0.989      0.986      0.712
                   a10        230         27          1      0.555      0.899      0.456
                    b1        230         20      0.972          1      0.995      0.793
                   aew        230         25      0.945          1       0.99      0.784
                   f22        230         17      0.913          1      0.995      0.725
                    p3        230        105       0.99          1      0.995      0.801
                    p8        230          1      0.637          1      0.995      0.597
                   f35        230         32      0.939      0.938      0.978      0.574
                   f18        230        125      0.985      0.992      0.987      0.817
                   v22        230         41      0.983          1      0.995       0.69
                 su-27        230         31      0.925          1      0.995      0.859
                 il-38        230         27      0.972          1      0.995      0.811
                tu-134        230          1      0.663          1      0.995      0.895
                 su-33        230          2          1      0.611      0.995      0.796
                 an-70        230          2      0.766          1      0.995       0.73
                 tu-22        230         98      0.984          1      0.995      0.831
Speed: 0.2ms preprocess, 3.8ms inference, 0.0ms loss, 0.8ms postprocess per image

改进一

测试结果

代码语言:javascript代码运行次数:0运行复制
YOLOv8l summary (fused): 376 layers, 53497878 parameters, 0 gradients, 369.2 GFLOPs
                 Class     Images  Instances      Box(P          R      mAP50  mAP50-95): 100%|██████████| 15/15 [00:03<00:00,  4.44it/s]
                   all        230       1412      0.951      0.935      0.985      0.738
                   c17        230        131      0.983      0.985      0.994      0.809
                    c5        230         68      0.971      0.979      0.993      0.831
            helicopter        230         43      0.972      0.953      0.971      0.612
                  c130        230         85          1       0.98      0.994      0.655
                   f16        230         57      0.827       0.93      0.916      0.658
                    b2        230          2      0.832          1      0.995      0.796
                 other        230         86      0.955      0.942      0.983      0.546
                   b52        230         70      0.928      0.916      0.958      0.806
                  kc10        230         62          1      0.984      0.988      0.827
               command        230         40      0.941          1      0.979      0.792
                   f15        230        123      0.991      0.924      0.978      0.693
                 kc135        230         91      0.961      0.956      0.961      0.679
                   a10        230         27          1      0.859      0.994      0.495
                    b1        230         20      0.971       0.95      0.993      0.688
                   aew        230         25      0.948          1      0.993      0.798
                   f22        230         17       0.96          1      0.995      0.772
                    p3        230        105      0.998          1      0.995      0.823
                    p8        230          1      0.955          1      0.995      0.597
                   f35        230         32      0.959      0.906       0.97      0.532
                   f18        230        125      0.965      0.976      0.987      0.808
                   v22        230         41      0.989          1      0.995      0.697
                 su-27        230         31      0.976          1      0.995      0.835
                 il-38        230         27      0.985          1      0.995      0.842
                tu-134        230          1      0.778          1      0.995      0.895
                 su-33        230          2          1          0      0.995      0.748
                 an-70        230          2      0.843          1      0.995      0.895
                 tu-22        230         98          1      0.995      0.995      0.791
Speed: 0.2ms preprocess, 7.8ms inference, 0.0ms loss, 0.7ms postprocess per image

改进二

测试结果

代码语言:javascript代码运行次数:0运行复制
YOLOv8l summary (fused): 318 layers, 44629302 parameters, 0 gradients, 165.8 GFLOPs
                 Class     Images  Instances      Box(P          R      mAP50  mAP50-95): 100%|██████████| 15/15 [00:02<00:00,  5.93it/s]
                   all        230       1412      0.951      0.955      0.985      0.724
                   c17        230        131      0.934      0.985      0.986      0.801
                    c5        230         68      0.947      0.912      0.987      0.811
            helicopter        230         43      0.974      0.874      0.958      0.562
                  c130        230         85      0.988      0.984      0.994      0.647
                   f16        230         57      0.908      0.871      0.941      0.649
                    b2        230          2      0.884          1      0.995      0.722
                 other        230         86      0.975      0.884      0.972      0.555
                   b52        230         70      0.944      0.967      0.968      0.801
                  kc10        230         62      0.997      0.984      0.989      0.835
               command        230         40          1      0.996      0.995      0.808
                   f15        230        123      0.979      0.943       0.99      0.688
                 kc135        230         91      0.979      0.967      0.988      0.674
                   a10        230         27          1      0.789       0.96      0.463
                    b1        230         20          1      0.992      0.995      0.718
                   aew        230         25      0.937          1      0.995      0.793
                   f22        230         17      0.887          1      0.982      0.715
                    p3        230        105      0.997          1      0.995      0.805
                    p8        230          1      0.793          1      0.995      0.597
                   f35        230         32          1      0.863      0.966      0.512
                   f18        230        125      0.955      0.992      0.985      0.811
                   v22        230         41      0.997          1      0.995      0.667
                 su-27        230         31      0.977          1      0.995      0.847
                 il-38        230         27      0.985          1      0.995      0.816
                tu-134        230          1      0.795          1      0.995      0.796
                 su-33        230          2          1      0.792      0.995      0.824
                 an-70        230          2       0.85          1      0.995      0.796
                 tu-22        230         98      0.996          1      0.995      0.825
Speed: 0.2ms preprocess, 4.4ms inference, 0.0ms loss, 0.7ms postprocess per image

出现了下降。说明增加检测头是有必要的!

总结

本文使用YoloV8复现了HIC-YOLOv5的改进,由于数据集中的物体并没有很多太小的!所以提升不明显!大家可以在自己的数据集上做尝试!

详见:

.2014.3001.5502

本文参与 腾讯云自媒体同步曝光计划,分享自微信公众号。原始发表:2023-10-30,如有侵权请联系 cloudcommunity@tencent 删除设计数据测试

本文标签: YoloV8改进策略复现HIC