YOLO系列模型改进指南

YOLO系列模型改进指南

目前包含yolov5，yolov7，yolov8模型的众多改进方案(都是B站视频教学和附带源码)，效果因数据集和参数而定，仅供参考。

如果需要改进模型，建议baseline和改进模型也不要载入预训练权重，不然的话，他们的起跑点不一样，没法做到公平对比。

目标检测教程视频指南大全(这是我的哔哩哔哩的视频整理合集，方便各位找到自己需要的视频教程，希望大家帮忙点点赞~)

Project

1. 基于整合yolov7的项目.项目详解
2. 基于YOLOV8改进的项目(内置90多种改进机制).项目详解
3. 基于YOLOV5,YOLOV7的(剪枝+知识蒸馏)项目.项目详解
4. 基于RT-DETR(Transformer类型的目标检测)的项目.项目详解
5. 基于YOLOV8的目标检测剪枝项目.项目详解
6. 基于YOLOV8的目标检测蒸馏项目.项目详解

改进指南

• iou

  添加EIOU，SIOU，ALPHA-IOU, FocalEIOU, Wise-IOU到yolov5,yolov8的box_iou中.

  1. yolov5

     视频教学地址：哔哩哔哩.

     博客地址：CSDN.

     2023-2-8 更新: 新增Wise-IoU 视频教学地址：哔哩哔哩. reference:github

  2. yolov8

     视频教学地址：哔哩哔哩.

     博客地址：CSDN.

     2023-2-7 更新: 新增Wise-IoU 视频教学地址：哔哩哔哩. reference:github

• yolov5-GFPN

  使用DAMO-YOLO中的GFPN替换YOLOV5中的Head.

  视频教学地址：哔哩哔哩.

• yolov5-C2F

  使用yolov8中的C2F模块替换yolov5中的C3模块.(这个操作比较简单，因此就不提供代码，直接看视频操作一下即可)

  视频教学地址：哔哩哔哩.

• yolov7-iou

  添加EIOU，SIOU，ALPHA-IOU, FocalEIOU, Wise-IOU到yolov7的box_iou中.

  视频教学地址：哔哩哔哩.

  博客地址：CSDN.

  2023-2-11 更新: 新增Wise-IoU 视频教学地址：哔哩哔哩. reference:github

• yolov5-OTA

  添加Optimal Transport Assignment到yolov5的Loss中.

  视频教学地址：哔哩哔哩.

• yolov5-DCN

  添加Deformable convolution V2到yolov5中.

  视频教学地址：哔哩哔哩.

• yolov8-DCN

  添加Deformable convolution V2到yolov8中.

  视频教学地址：哔哩哔哩.

• yolov7-DCN

  添加Deformable convolution V2到yolov7中.

  视频教学地址：哔哩哔哩.

• yolov5-AUX

  添加辅助训练分支到yolov5中.

  视频教学地址：哔哩哔哩.

  原理参考链接：知乎

• CAM

  添加context augmentation module到yolov5中.

  视频教学地址：哔哩哔哩.

  paper：链接

• yolov5-SAConv

  添加SAC到yolov5中.

  视频教学地址：哔哩哔哩.

  paper：链接

  reference: 链接

• yolov7-SAConv

  添加SAC到yolov7中.

  视频教学地址：哔哩哔哩.

  paper：链接

  reference: 链接

• yolov5-CoordConv

  添加CoordConv到yolov5中.

  视频教学地址：哔哩哔哩.

  reference: 链接

• yolov5-soft-nms

  添加soft-nms(IoU,GIoU,DIoU,CIoU,EIoU,SIoU)到yolov5中.

  视频教学地址：哔哩哔哩.

• yolov7-CoordConv

  添加CoordConv到yolov7中.

  视频教学地址：哔哩哔哩.

  reference: 链接

• yolov7-soft-nms

  添加soft-nms(IoU,GIoU,DIoU,CIoU,EIoU,SIoU)到yolov7中.

  视频教学地址：哔哩哔哩.

• yolov5-DSConv

  添加DSConv到yolov5中.

  视频教学地址：哔哩哔哩.

  paper: 链接

  reference: 链接

• yolov7-DSConv

  添加DSConv到yolov7中.

  视频教学地址：哔哩哔哩.

  paper: 链接

  reference: 链接

• yolov5-DCNV3

  添加DCNV3到yolov5中.

  视频教学地址：哔哩哔哩.

  补充事项-视频教学地址：哔哩哔哩.

  paper: 链接

  reference: 链接

• yolov5-NWD

  添加Normalized Gaussian Wasserstein Distance到yolov5中.

  视频教学地址：哔哩哔哩.

  paper: 链接

  reference: 链接

• yolov7-DCNV3

  添加DCNV3到yolov7中.

  视频教学地址：哔哩哔哩.

  paper: 链接

  reference: 链接

• yolov7-DCNV3

  添加DCNV3到yolov7中.

  视频教学地址：哔哩哔哩.

  paper: 链接

  reference: 链接

• yolov5-DecoupledHead

  添加Efficient-DecoupledHead到yolov5中.

  视频教学地址：哔哩哔哩.

  paper: yolov6链接

  reference: 链接

• yolov5-FasterBlock

  添加FasterNet中的Faster-Block到yolov5中.

  视频教学地址：哔哩哔哩.

  paper: 链接

  reference: 链接

• yolov7-NWD

  添加Normalized Gaussian Wasserstein Distance到yolov7中.

  视频教学地址：哔哩哔哩.

  paper: 链接

  reference: 链接

• yolov7-DecoupledHead

  添加具有隐式知识学习的Efficient-DecoupledHead到yolov7中.

  视频教学地址：哔哩哔哩.

  paper: yolov6链接 yolor链接 yolor参考博客

  reference: 链接

• yolov5-backbone

  添加Timm支持的主干到yolov5中.

  需要安装timm库. 命令: pip install -i https://pypi.tuna.tsinghua.edu.cn/simple timm

  视频教学地址：哔哩哔哩.

  reference: 链接

• yolov7-PConv

  添加FasterNet中的PConv到yolov7中.

  视频教学地址：哔哩哔哩.

  paper: 链接

  reference: 链接

• yolov5-TSCODE

  添加Task-Specific Context Decoupling到yolov5中.

  需要安装einops库. 命令: pip install -i https://pypi.tuna.tsinghua.edu.cn/simple einops

  视频教学地址：哔哩哔哩.

  paper: yolov6链接

• yolov5-backbone/fasternet

  添加FasterNet主干到yolov5中.

  视频教学地址：哔哩哔哩.

  reference: 链接

• yolov5-backbone/ODConv

  添加Omni-Dimensional Dynamic Convolution主干(od_mobilenetv2,od_resnet)到yolov5中.

  视频教学地址：哔哩哔哩.

  reference: 链接

• yolov5-backbone/ODConvFuse

  融合Omni-Dimensional Dynamic Convolution主干(od_mobilenetv2,od_resnet)中的Conv和BN.

  视频教学地址：哔哩哔哩.

• yolov5-CARAFE

  添加轻量级上采样算子CARAFE到yolov5中.

  视频教学地址：哔哩哔哩. yolov7修改视频-哔哩哔哩.

  reference: 链接

• yolov5-EVC

  添加CFPNet中的EVC-Block到yolov5中.

  视频教学地址：哔哩哔哩.

  reference: 链接

• yolov5-dyhead

  添加基于注意力机制的目标检测头(DYHEAD)到yolov5中.

  yolov7版本: 哔哩哔哩.

  安装命令:

    pip install -U openmim
    mim install mmengine
    mim install "mmcv>=2.0.0"
  

  视频教学地址：哔哩哔哩.

  reference: 链接

  paper: 链接

• yolov5-backbone/inceptionnext

  添加(2023年New)InceptionNeXt主干到yolov5中.

  视频教学地址：哔哩哔哩.

  reference: 链接

  paper: 链接

• yolov5-aLRPLoss

  添加aLRPLoss到yolov5中.

  视频教学地址：哔哩哔哩.

  reference: 链接

  paper: 链接

• yolov5-res2block

  结合Res2Net提出具有多尺度提取能力的C3模块.

  视频教学地址：哔哩哔哩.

  reference: 链接

  paper: 链接

• yolov7-odconv

  添加Omni-Dimensional Dynamic Convolution到yolov7中.

  视频教学地址：哔哩哔哩.

  reference: 链接

• yolov5-backbone/FocalNet

  添加(2022年)FocalNet(transformer)主干到yolov5中.

  视频教学地址：哔哩哔哩.

  reference: 链接

  paper: 链接

• yolov5-backbone/EMO

  添加(2023年)EMO(transformer)主干到yolov5中.

  视频教学地址：哔哩哔哩.

  reference: 链接

  paper: 链接

• yolov5-backbone/EfficientFormerV2

  添加(2022年)EfficientFormerV2(transformer)主干到yolov5中.

  视频教学地址：哔哩哔哩.

  reference: 链接

  paper: 链接

  weight_download: 百度网盘链接

• yolov5-backbone/PoolFormer

  添加(2022年CVPR)PoolFormer(transformer)主干到yolov5中.

  视频教学地址：哔哩哔哩.

  reference: 链接

  paper: 链接

• yolov5-backbone/EfficientViT

  添加(2023年)EfficientViT(transformer)主干到yolov5中.

  视频教学地址：哔哩哔哩.

  reference: 链接

  paper: 链接

  weight_download: 百度网盘链接

• yolov5-ContextAggregation

  添加ContextAggregation到yolov5中.

  视频教学地址：哔哩哔哩.

  reference: 链接

  paper: 链接

• yolov5-backbone/VanillaNet

  添加(2023年)VanillaNet主干到yolov5中.

  视频教学地址：哔哩哔哩.

  reference: 链接

  paper: 链接

  weight_download: 百度网盘链接

• yolov7-EVC

  添加CFPNet中的EVC-Block到yolov7中.

  视频教学地址：哔哩哔哩.

  reference: 链接

• yolov7-head

  P2,P6检测层在YOLOV7中的添加.

  视频教学地址：哔哩哔哩.

• yolov7-slimneck

  使用VOVGSCSP轻量化yolov7的Neck.

  视频教学地址：哔哩哔哩.

  reference: 链接

• yolov5-SwinTransformer

  添加SwinTransformer-Tiny主干到yolov5中.

  视频教学地址：哔哩哔哩.

  reference: 链接

  weight_download: SwinTransformer-Tiny百度链接

• yolov5-NextViT

  添加(2022年)NextViT主干到yolov5中.

  视频教学地址：哔哩哔哩.

  reference: 链接

  weight_download: SwinTransformer-Tiny百度链接

• yolov5-ConvNextV2

  添加(2023年)ConvNextV2主干到yolov5中.

  视频教学地址：哔哩哔哩.

  reference: 链接

• yolov5-RIFormer

  添加(2023年)RIFormer主干到yolov5中.

  视频教学地址：哔哩哔哩.

  reference: mmpretrain链接

  weight_download: mmpretrain链接

• yolov5-C3RFEM

  Scale-Aware RFE与C3结合而成的C3RFEM添加到yolov5中.

  视频教学地址：哔哩哔哩.

  reference: 链接

• yolov7-RFEM

  Scale-Aware RFE添加到yolov7中.

  视频教学地址：哔哩哔哩.

  reference: 链接

• yolov5-DBB

  把重参数结构DiverseBranchBlock与C3融合成C3-DBB添加到yolov5中.

  视频教学地址：哔哩哔哩.

  reference: 链接

• yolov7-DBB

  把重参数结构DiverseBranchBlock添加到yolov7中.

  视频教学地址：哔哩哔哩.

  reference: 链接

• yolov5-backbone/CVPR2023-EfficientViT

  添加(2023CVPR)EfficientViT(transformer)主干到yolov5中.

  视频教学地址：哔哩哔哩.

  reference: 链接

  paper: 链接

  weight: github链接

• yolov5-backbone/LSKNet

  添加(2023旋转目标检测SOTA)LSKNet主干到yolov5中.

  视频教学地址：哔哩哔哩.

  reference: 链接

  paper: 链接

• yolov5-MPDiou

  添加(2023最新IoU度量算法)MPDiou到yolov5中.(视频教学地址中为详细从头手把手教学,因此本项没有提供代码)

  视频教学地址：哔哩哔哩.

  paper: 链接

• yolov7-MPDiou

  添加(2023最新IoU度量算法)MPDiou到yolov7中.

  视频教学地址：哔哩哔哩.

  paper: 链接

• yolov5-SlideLoss

  添加Yolo-Face-V2中SlideLoss的到yolov5中.

  视频教学地址：哔哩哔哩.

  reference: 链接

  paper: 链接

• yolov5-backbone/CVPR2023-RepViT

  添加RepViT(transformer)主干到yolov5中.

  视频教学地址：哔哩哔哩.

  reference: 链接

  paper: 链接

• yolov5-GOLDYOLO

  利用华为2023最新GOLD-YOLO中的Gatherand-Distribute进行改进YOLOV5中的特征融合模块.

  视频教学地址：哔哩哔哩.

  reference: 链接

  paper: 链接

• yolov7-GOLDYOLO(文件在yolov5-GOLDYOLO的文件夹中)

  利用华为2023最新GOLD-YOLO中的Gatherand-Distribute进行改进YOLOV7中的特征融合模块.

  视频教学地址：哔哩哔哩.

  reference: 链接

  paper: 链接

• yolov5-DySnakeConv

  利用动态蛇形卷积改进YOLOV5.

  视频教学地址：哔哩哔哩.

  reference: 链接

  paper: 链接

• yolov7-DySnakeConv

  利用动态蛇形卷积改进YOLOV7.

  视频教学地址：哔哩哔哩.

  reference: 链接

  paper: 链接

• yolov5-AIFI

  利用带有位置信息编码的AIFI自注意力机制改进YOLOV5.

  视频教学地址：哔哩哔哩.

  reference: 链接

  paper: 链接

• yolov7-AIFI

  利用带有位置信息编码的AIFI自注意力机制改进YOLOV7.

  视频教学地址：哔哩哔哩.

  reference: 链接

  paper: 链接

• yolov5-backbone/UniRepLKNet

  添加UniRepLKNet主干到yolov5中.

  视频教学地址：哔哩哔哩.

  reference: 链接

  paper: 链接

  weights-download: 百度云链接

• yolov5-asf

  添加Attentional Scale Sequence Fusion到yolov5中.

  视频教学地址：哔哩哔哩.

  reference: 链接

  paper: 链接

• yolov5-ccfm

  添加cross-scale feature-fusion到yolov5中.

  视频教学地址：哔哩哔哩.

  reference: 链接

  paper: 链接

• yolov7-asf

  添加Attentional Scale Sequence Fusion到yolov7中.

  视频教学地址：哔哩哔哩.

  reference: 链接

  paper: 链接

• yolov5-RepNCSPELAN

  添加yolov9中的RepNCSPELAN到yolov5中.

  视频教学地址：哔哩哔哩.

  reference: 链接

  paper: 链接

• yolov7-RepNCSPELAN

  添加yolov9中的RepNCSPELAN到yolov7中.

  视频教学地址：哔哩哔哩.

  reference: 链接

  paper: 链接