Skip to content

Latest commit

 

History

History
362 lines (357 loc) · 22.3 KB

File metadata and controls

362 lines (357 loc) · 22.3 KB

YOLO-Improve

这个项目主要是提供一些关于yolo系列模型的改进思路,效果因数据集和参数而异,仅作参考。

Explanation

  • 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: 链接

  • 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: 百度云链接

  • 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: 链接

  • yolov9-backbone
    添加各种backbone到yolov9中.
    视频教学地址:哔哩哔哩.

  • yolov5-backbone/CVPR2024-StarNet
    添加CVPR2024-StarNet到yolov5、yolov7、yolov9中.
    视频教学地址:哔哩哔哩.