本文主要是介绍RTDETR改进系列指南,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
基于Ultralytics的RT-DETR改进项目.(89.9¥)
为了感谢各位对RTDETR项目的支持,本项目的赠品是yolov5-PAGCP通道剪枝算法.具体使用教程
自带的一些文件说明
- train.py
训练模型的脚本 - main_profile.py
输出模型和模型每一层的参数,计算量的脚本(rtdetr-l和rtdetr-x因为thop库的问题,没办法正常输出每一层的参数和计算量和时间) - val.py
使用训练好的模型计算指标的脚本 - detect.py
推理的脚本 - track.py
跟踪推理的脚本 - heatmap.py
生成热力图的脚本 - get_FPS.py
计算模型储存大小、模型推理时间、FPS的脚本 - get_COCO_metrice.py
计算COCO指标的脚本 - plot_result.py
绘制曲线对比图的脚本
RT-DETR基准模型
-
ultralytics/cfg/models/rt-detr/rtdetr-r18.yaml(有预训练权重COCO+Objects365,来自RTDETR-Pytorch版本的移植)
rtdetr-r18 summary: 421 layers, 20184464 parameters, 20184464 gradients, 58.6 GFLOPs
-
ultralytics/cfg/models/rt-detr/rtdetr-r34.yaml(有预训练权重COCO,来自RTDETR-Pytorch版本的移植)
rtdetr-r34 summary: 525 layers, 31441668 parameters, 31441668 gradients, 90.6 GFLOPs
-
ultralytics/cfg/models/rt-detr/rtdetr-r50-m.yaml(有预训练权重COCO,来自RTDETR-Pytorch版本的移植)
rtdetr-r50-m summary: 637 layers, 36647020 parameters, 36647020 gradients, 98.3 GFLOPs
-
ultralytics/cfg/models/rt-detr/rtdetr-r50.yaml(有预训练权重COCO+Objects365,来自RTDETR-Pytorch版本的移植)
rtdetr-r50 summary: 629 layers, 42944620 parameters, 42944620 gradients, 134.8 GFLOPs
-
ultralytics/cfg/models/rt-detr/rtdetr-r101.yaml
rtdetr-r101 summary: 867 layers, 76661740 parameters, 76661740 gradients, 257.7 GFLOPs
-
ultralytics/cfg/models/rt-detr/rtdetr-l.yaml(有预训练权重)
rtdetr-l summary: 673 layers, 32970732 parameters, 32970732 gradients, 108.3 GFLOPs
-
ultralytics/cfg/models/rt-detr/rtdetr-x.yaml(有预训练权重)
rtdetr-x summary: 867 layers, 67468108 parameters, 67468108 gradients, 232.7 GFLOPs
专栏改进汇总
二次创新系列
-
ultralytics/cfg/models/rt-detr/rtdetr-DCNV2-Dynamic.yaml
使用自研可变形卷积DCNV2-Dynamic改进resnet18-backbone中的BasicBlock.(详细介绍请看百度云视频-MPCA与DCNV2_Dynamic的说明)
-
ultralytics/cfg/models/rt-detr/rtdetr-iRMB-Cascaded.yaml
使用EfficientViT CVPR2023中的CascadedGroupAttention对EMO ICCV2023中的iRMB进行二次创新来改进resnet18-backbone中的BasicBlock.(详细介绍请看百度云视频-20231119更新说明)
-
ultralytics/cfg/models/rt-detr/rtdetr-PConv-Rep.yaml
使用RepVGG CVPR2021中的RepConv对FasterNet CVPR2023中的PConv进行二次创新后改进resnet18-backbone中的BasicBlock.
-
ultralytics/cfg/models/rt-detr/rtdetr-Faster-Rep.yaml
使用RepVGG CVPR2021中的RepConv对FasterNet CVPR2023中的Faster-Block进行二次创新后改进resnet18-backbone中的BasicBlock.
-
ultralytics/cfg/models/rt-detr/rtdetr-Faster-EMA.yaml
使用EMA ICASSP2023对FasterNet CVPR2023中的Faster-Block进行二次创新后改进resnet18-backbone中的BasicBlock.
-
ultralytics/cfg/models/rt-detr/rtdetr-Faster-Rep-EMA.yaml
使用RepVGG CVPR2021中的RepConv和EMA ICASSP2023对FasterNet CVPR2023中的Faster-Block进行二次创新后改进resnet18-backbone中的BasicBlock.
-
ultralytics/cfg/models/rt-detr/rtdetr-DWRC3-DRB.yaml
使用UniRepLKNet中的DilatedReparamBlock对DWRSeg中的Dilation-wise Residual(DWR)进行二次创新改进rtdetr.
-
ultralytics/cfg/models/rt-detr/rtdetr-ASF-P2.yaml
在ultralytics/cfg/models/rt-detr/rtdetr-ASF.yaml的基础上进行二次创新,引入P2检测层并对网络结构进行优化.
-
ultralytics/cfg/models/rt-detr/rtdetr-slimneck-ASF.yaml
使用SlimNeck中的VoVGSCSP\VoVGSCSPC和GSConv和ASF-YOLO中的Attentional Scale Sequence Fusion改进rtdetr中的CCFM.
-
ultralytics/cfg/models/rt-detr/rtdetr-goldyolo-asf.yaml
利用华为2023最新GOLD-YOLO中的Gatherand-Distribute和ASF-YOLO中的Attentional Scale Sequence Fusion进行改进特征融合模块.
-
ultralytics/cfg/models/rt-detr/rtdetr-HSPAN.yaml
对MFDS-DETR中的HS-FPN进行二次创新后得到HSPAN改进RTDETR中的CCFM.
-
ultralytics/cfg/models/rt-detr/rtdetr-ASF-Dynamic.yaml
使用ICCV2023 DySample改进ASF-YOLO中的Attentional Scale Sequence Fusion的上采样模块得到Dynamic Sample Attentional Scale Sequence Fusion改进CCFM.
-
ultralytics/cfg/models/rt-detr/rtdetr-iRMB-DRB.yaml
使用UniRepLKNet中的DilatedReparamBlock对EMO ICCV2023中的iRMB进行二次创新来改进resnet18-backbone中的BasicBlock.
-
ultralytics/cfg/models/rt-detr/rtdetr-iRMB-SWC.yaml
使用shift-wise conv对EMO ICCV2023中的iRMB进行二次创新来改进resnet18-backbone中的BasicBlock.
自研系列
待更新
BackBone系列
-
ultralytics/cfg/models/rt-detr/rt-detr-timm.yaml
使用timm库系列的主干替换rtdetr的backbone.(基本支持现有CNN模型)
-
ultralytics/cfg/models/rt-detr/rt-detr-fasternet.yaml
使用FasterNet CVPR2023替换rtdetr的backbone.
-
ultralytics/cfg/models/rt-detr/rt-detr-EfficientViT.yaml
使用EfficientViT CVPR2023替换rtdetr的backbone.
-
ultralytics/cfg/models/rt-detr/rtdetr-convnextv2.yaml
使用ConvNextV2 2023替换rtdetr的backbone.
-
ultralytics/cfg/models/rt-detr/rtdetr-EfficientFormerv2.yaml
使用EfficientFormerv2 2022替换rtdetr的backbone.
-
ultralytics/cfg/models/rt-detr/rtdetr-repvit.yaml
使用RepViT ICCV2023替换rtdetr的backbone.
-
ultralytics/cfg/models/rt-detr/rtdetr-CSwomTramsformer.yaml
使用CSwinTramsformer CVPR2022替换rtdetr的backbone.
-
ultralytics/cfg/models/rt-detr/rtdetr-VanillaNet.yaml
使用VanillaNet 2023替换rtdetr的backbone.
-
ultralytics/cfg/models/rt-detr/rtdetr-SwinTransformer.yaml
使用SwinTransformer ICCV2021替换rtdetr的backbone.
-
ultralytics/cfg/models/rt-detr/rtdetr-lsknet.yaml
使用LSKNet ICCV2023替换rtdetr的backbone.
-
ultralytics/cfg/models/rt-detr/rt-detr-unireplknet.yaml
使用UniRepLKNet替换rtdetr的backbone.
-
ultralytics/cfg/models/rt-detr/rtdetr-TransNeXt.yaml
使用TransNeXt改进rtdetr的backbone.
AIFI系列
-
ultralytics/cfg/models/rt-detr/rtdetr-AIFI-LPE.yaml
使用LearnedPositionalEncoding改进AIFI中的位置编码生成.(详细介绍请看百度云视频-20231119更新说明)
-
ultralytics/cfg/models/rt-detr/rtdetr-CascadedGroupAttention.yaml
使用EfficientViT CVPR2023中的CascadedGroupAttention改进rtdetr中的AIFI.(详细请看百度云视频-rtdetr-CascadedGroupAttention说明)
-
ultralytics/cfg/models/rt-detr/rtdetr-AIFI-DAttention.yaml
使用Vision Transformer with Deformable Attention CVPR2022中的DAttention改进AIFI.
-
ultralytics/cfg/models/rt-detr/rtdetr-AIFI-HiLo.yaml
使用LITv2中具有提取高低频信息的高效注意力对AIFI进行二次改进.
Neck系列
-
ultralytics/cfg/models/rt-detr/rtdetr-ASF.yaml
使用ASF-YOLO中的Attentional Scale Sequence Fusion来改进rtdetr.
-
ultralytics/cfg/models/rt-detr/rtdetr-slimneck.yaml
使用SlimNeck中的VoVGSCSP\VoVGSCSPC和GSConv改进rtdetr中的CCFM.
-
ultralytics/cfg/models/rt-detr/rtdetr-SDI.yaml
使用U-NetV2中的 Semantics and Detail Infusion Module对CCFM中的feature fusion进行改进.
-
ultralytics/cfg/models/rt-detr/rtdetr-goldyolo.yaml
利用华为2023最新GOLD-YOLO中的Gatherand-Distribute进行改进特征融合模块.
-
ultralytics/cfg/models/rt-detr/rtdetr-HSFPN.yaml
使用MFDS-DETR中的HS-FPN改进RTDETR中的CCFM.
Head系列
-
ultralytics/cfg/models/rt-detr/rtdetr-p2.yaml
添加小目标检测头P2到TransformerDecoderHead中.
RepC3改进系列
-
ultralytics/cfg/models/rt-detr/rtdetr-DWRC3.yaml
使用DWRSeg中的Dilation-wise Residual(DWR)模块构建DWRC3改进rtdetr.
-
ultralytics/cfg/models/rt-detr/rtdetr-Conv3XCC3.yaml
使用Swift Parameter-free Attention Network中的Conv3XC改进RepC3.
-
ultralytics/cfg/models/rt-detr/rtdetr-DRBC3.yaml
使用UniRepLKNet中的DilatedReparamBlock改进RepC3.
-
ultralytics/cfg/models/rt-detr/rtdetr-DBBC3.yaml
使用DiverseBranchBlock CVPR2021改进RepC3.
ResNet主干中的BasicBlock/BottleNeck改进系列(以下改进BottleNeck基本都有,就不再重复标注)
-
ultralytics/cfg/models/rt-detr/rtdetr-Ortho.yaml
使用OrthoNets中的正交通道注意力改进resnet18-backbone中的BasicBlock.(详细介绍请看百度云视频-20231119更新说明)
-
ultralytics/cfg/models/rt-detr/rtdetr-DCNV2.yaml
使用可变形卷积DCNV2改进resnet18-backbone中的BasicBlock.
-
ultralytics/cfg/models/rt-detr/rtdetr-DCNV3.yaml
使用可变形卷积DCNV3 CVPR2023改进resnet18-backbone中的BasicBlock.(安装教程请看百度云视频-20231119更新说明)
-
ultralytics/cfg/models/rt-detr/rtdetr-iRMB.yaml
使用EMO ICCV2023中的iRMB改进resnet18-backbone中的BasicBlock.(详细介绍请看百度云视频-20231119更新说明)
-
ultralytics/cfg/models/rt-detr/rtdetr-DySnake.yaml
添加DySnakeConv到resnet18-backbone中的BasicBlock中.
-
ultralytics/cfg/models/rt-detr/rtdetr-PConv.yaml
使用FasterNet CVPR2023中的PConv改进resnet18-backbone中的BasicBlock.
-
ultralytics/cfg/models/rt-detr/rtdetr-Faster.yaml
使用FasterNet CVPR2023中的Faster-Block改进resnet18-backbone中的BasicBlock.
-
ultralytics/cfg/models/rt-detr/rtdetr-AKConv.yaml
使用AKConv 2023改进resnet18-backbone中的BasicBlock.
-
ultralytics/cfg/models/rt-detr/rtdetr-RFAConv.yaml
使用RFAConv 2023改进resnet18-backbone中的BasicBlock.
-
ultralytics/cfg/models/rt-detr/rtdetr-RFCAConv.yaml
使用RFCAConv 2023改进resnet18-backbone中的BasicBlock.
-
ultralytics/cfg/models/rt-detr/rtdetr-RFCBAMConv.yaml
使用RFCBAMConv 2023改进resnet18-backbone中的BasicBlock.
-
ultralytics/cfg/models/rt-detr/rtdetr-Conv3XC.yaml
使用Swift Parameter-free Attention Network中的Conv3XC改进resnet18-backbone中的BasicBlock.
-
ultralytics/cfg/models/rt-detr/rtdetr-DRB.yaml
使用UniRepLKNet中的DilatedReparamBlock改进resnet18-backbone中的BasicBlock.
-
ultralytics/cfg/models/rt-detr/rtdetr-DBB.yaml
使用DiverseBranchBlock CVPR2021改进resnet18-backbone中的BasicBlock.
-
ultralytics/cfg/models/rt-detr/rtdetr-DualConv.yaml
使用DualConv改进resnet18-backbone中的BasicBlock.
-
ultralytics/cfg/models/rt-detr/rtdetr-AggregatedAtt.yaml
使用TransNeXt中的聚合感知注意力改进resnet18中的BasicBlock.(百度云视频-20240106更新说明)
-
ultralytics/cfg/models/rt-detr/rtdetr-DCNV4.yaml
使用DCNV4改进resnet18中的BasicBlock.
-
ultralytics/cfg/models/rt-detr/rtdetr-SWC.yaml
使用shift-wise conv改进resnet18中的BasicBlock.
上下采样算子系列
-
ultralytics/cfg/models/rt-detr/rtdetr-DySample.yaml
使用ICCV2023 DySample改进CCFM中的上采样.
-
ultralytics/cfg/models/rt-detr/rtdetr-CARAFE.yaml
使用ICCV2019 CARAFE改进CCFM中的上采样.
-
ultralytics/cfg/models/rt-detr/rtdetr-HWD.yaml
使用Haar wavelet downsampling改进CCFM的下采样.
RT-DETR-L改进系列
-
ultralytics/cfg/models/rt-detr/rtdetr-l-GhostHGNetV2.yaml
使用GhostConv改进HGNetV2.(详细介绍请看百度云视频-20231109更新说明)
-
ultralytics/cfg/models/rt-detr/rtdetr-l-RepHGNetV2.yaml
使用RepConv改进HGNetV2.(详细介绍请看百度云视频-20231109更新说明)
-
ultralytics/cfg/models/rt-detr/rtdetr-l-attention.yaml
添加注意力模块到HGBlock中.(手把手教程请看百度云视频-手把手添加注意力教程)
注意力系列
- EMA
- SimAM
- SpatialGroupEnhance
- BiLevelRoutingAttention, BiLevelRoutingAttention_nchw
- TripletAttention
- CoordAtt
- CBAM
- BAMBlock
- EfficientAttention(CloFormer中的注意力)
- LSKBlock
- SEAttention
- CPCA
- deformable_LKA
- EffectiveSEModule
- LSKA
- SegNext_Attention
- DAttention(Vision Transformer with Deformable Attention CVPR2022)
- FocusedLinearAttention(ICCV2023)
- MLCA
- TransNeXt_AggregatedAttention
- HiLo
IoU系列
- IoU,GIoU,DIoU,CIoU,EIoU,SIoU(百度云视频-20231125更新说明)
- MPDIoU论文链接(百度云视频-20231125更新说明)
- Inner-IoU,Inner-GIoU,Inner-DIoU,Inner-CIoU,Inner-EIoU,Inner-SIoU论文链接(百度云视频-20231125更新说明)
- Inner-MPDIoU(利用Inner-Iou与MPDIou进行二次创新)(百度云视频-20231125更新说明)
- Normalized Gaussian Wasserstein Distance.论文链接(百度云视频-20231125更新说明)
- Shape-IoU,Inner-Shape-IoU论文链接(百度云视频-20240106更新说明)
- SlideLoss,EMASlideLoss创新思路.Yolo-Face V2(百度云视频-20240113更新说明)
- Wise-IoU(v1,v2,v3)系列(IoU,WIoU,EIoU,GIoU,DIoU,CIoU,SIoU,MPDIoU,ShapeIoU)(百度云视频-20240113更新说明)
- Inner-Wise-IoU(v1,v2,v3)系列(IoU,WIoU,EIoU,GIoU,DIoU,CIoU,SIoU,MPDIoU,ShapeIoU)(百度云视频-20240113更新说明)
- Focaler-IoU,Focaler-GIoU,Focaler-DIoU,Focaler-CIoU,Focaler-EIoU,Focaler-SIoU,Focaler-Shape-IoU,Focaler-MPDIoU论文链接(百度云视频-20240128更新说明)
- Focaler-Wise-IoU(v1,v2,v3)(IoU,WIoU,EIoU,GIoU,DIoU,CIoU,SIoU,MPDIoU,ShapeIoU)论文链接(百度云视频-20240128更新说明)
以Yolov8为基准模型的改进方案
-
ultralytics/cfg/models/yolo-detr/yolov8-detr.yaml
使用RT-DETR中的TransformerDecoderHead改进yolov8.
-
ultralytics/cfg/models/yolo-detr/yolov8-detr-DWR.yaml
使用RT-DETR中的TransformerDecoderHead和DWRSeg中的Dilation-wise Residual(DWR)模块改进yolov8.
-
ultralytics/cfg/models/yolo-detr/yolov8-detr-fasternet.yaml
使用RT-DETR中的TransformerDecoderHead和FasterNet CVPR2023改进yolov8.(支持替换其他主干,请看百度云视频-替换主干示例教程)
-
ultralytics/cfg/models/yolo-detr/yolov8-detr-AIFI-LPE.yaml
使用RT-DETR中的TransformerDecoderHead和LearnedPositionalEncoding改进yolov8.(详细介绍请看百度云视频-20231119更新说明)
-
ultralytics/cfg/models/yolo-detr/yolov8-detr-C2f-DCNV2.yaml
使用RT-DETR中的TransformerDecoderHead和可变形卷积DCNV2改进yolov8.
-
ultralytics/cfg/models/yolo-detr/yolov8-detr-C2f-DCNV3.yaml
使用RT-DETR中的TransformerDecoderHead和可变形卷积DCNV3 CVPR2023改进yolov8.(安装教程请看百度云视频-20231119更新说明)
-
ultralytics/cfg/models/yolo-detr/yolov8-detr-C2f-DCNV2-Dynamic.yaml
使用RT-DETR中的TransformerDecoderHead和自研可变形卷积DCNV2-Dynamic改进yolov8.(详细介绍请看百度云视频-MPCA与DCNV2_Dynamic的说明)
-
ultralytics/cfg/models/yolo-detr/yolov8-detr-C2f-Ortho.yaml
使用RT-DETR中的TransformerDecoderHead和OrthoNets中的正交通道注意力改进yolov8.(详细介绍请看百度云视频-20231119更新说明)
-
ultralytics/cfg/models/yolo-detr/yolov8-detr-attention.yaml
添加注意力到基于RTDETR-Head中的yolov8中.(手把手教程请看百度云视频-手把手添加注意力教程)
-
ultralytics/cfg/models/yolo-detr/yolov8-detr-p2.yaml
添加小目标检测头P2到TransformerDecoderHead中.
-
ultralytics/cfg/models/yolo-detr/yolov8-detr-C2f-DySnake.yaml
DySnakeConv与C2f融合.
-
ultralytics/cfg/models/yolo-detr/yolov8-detr-C2f-Faster.yaml
使用RT-DETR中的TransformerDecoderHead和FasterNet CVPR2023中的Faster-Block改进yolov8.
-
ultralytics/cfg/models/yolo-detr/yolov8-detr-C2f-Faster-Rep.yaml
使用RT-DETR中的TransformerDecoderHead和FasterNet CVPR2023中与RepVGG CVPR2021中的RepConv二次创新后的Faster-Block-Rep改进yolov8.
-
ultralytics/cfg/models/yolo-detr/yolov8-detr-C2f-Faster-EMA.yaml
使用RT-DETR中的TransformerDecoderHead和FasterNet CVPR2023中与EMA ICASSP2023二次创新后的Faster-Block-EMA的Faster-Block-EMA改进yolov8.
-
ultralytics/cfg/models/yolo-detr/yolov8-detr-C2f-Faster-Rep-EMA.yaml
使用RT-DETR中的TransformerDecoderHead和FasterNet CVPR2023中与RepVGG CVPR2021中的RepConv、EMA ICASSP2023二次创新后的Faster-Block改进yolov8.
-
ultralytics/cfg/models/yolo-detr/yolov8-detr-C2f-AKConv.yaml
使用RT-DETR中的TransformerDecoderHead和AKConv 2023改进yolov8.
-
ultralytics/cfg/models/yolo-detr/yolov8-detr-C2f-RFAConv.yaml
使用RT-DETR中的TransformerDecoderHead和RFAConv 2023改进yolov8.
-
ultralytics/cfg/models/yolo-detr/yolov8-detr-C2f-RFAConv.yaml
使用RT-DETR中的TransformerDecoderHead和RFCAConv 2023改进yolov8.
-
ultralytics/cfg/models/yolo-detr/yolov8-detr-C2f-RFAConv.yaml
使用RT-DETR中的TransformerDecoderHead和RFCBAMConv 2023改进yolov8.
-
ultralytics/cfg/models/yolo-detr/yolov8-detr-C2f-Conv3XC.yaml
使用RT-DETR中的TransformerDecoderHead和Swift Parameter-free Attention Network中的Conv3XC改进yolov8.
-
ultralytics/cfg/models/yolo-detr/yolov8-detr-C2f-SPAB.yaml
使用RT-DETR中的TransformerDecoderHead和Swift Parameter-free Attention Network中的SPAB改进yolov8.
-
ultralytics/cfg/models/yolo-detr/yolov8-detr-C2f-DRB.yaml
使用RT-DETR中的TransformerDecoderHead和UniRepLKNet中的DilatedReparamBlock改进yolov8.
-
ultralytics/cfg/models/yolo-detr/yolov8-detr-C2f-UniRepLKNetBlock.yaml
使用RT-DETR中的TransformerDecoderHead和UniRepLKNet中的UniRepLKNetBlock改进yolov8.
-
ultralytics/cfg/models/yolo-detr/yolov8-detr-DWR-DRB.yaml
使用UniRepLKNet中的DilatedReparamBlock对DWRSeg中的Dilation-wise Residual(DWR)进行二次创新改进yolov8.
-
ultralytics/cfg/models/yolo-detr/yolov8-detr-C2f-DBB.yaml
使用RT-DETR中的TransformerDecoderHead和DiverseBranchBlock CVPR2021改进yolov8.
-
ultralytics/cfg/models/yolo-detr/yolov8-detr-CSP-EDLAN.yaml
使用RT-DETR中的TransformerDecoderHead和DualConv打造CSP Efficient Dual Layer Aggregation Networks改进yolov8.
-
ultralytics/cfg/models/yolo-detr/yolov8-detr-ASF.yaml
使用RT-DETR中的TransformerDecoderHead和ASF-YOLO中的Attentional Scale Sequence Fusion改进yolov8.
-
ultralytics/cfg/models/yolo-detr/yolov8-detr-ASF-P2.yaml
在ultralytics/cfg/models/yolo-detr/yolov8-detr-ASF.yaml的基础上进行二次创新,引入P2检测层并对网络结构进行优化.
-
ultralytics/cfg/models/yolo-detr/yolov8-detr-slimneck.yaml
使用RT-DETR中的TransformerDecoderHead和SlimNeck中VoVGSCSP\VoVGSCSPC和GSConv改进yolov8的neck.
-
ultralytics/cfg/models/yolo-detr/yolov8-detr-slimneck-asf.yaml
在ultralytics/cfg/models/yolo-detr/yolov8-detr-slimneck.yaml使用ASF-YOLO中的Attentional Scale Sequence Fusion进行二次创新.
-
ultralytics/cfg/models/yolo-detr/yolov8-detr-C2f-AggregatedAtt.yaml
使用RT-DETR中的TransformerDecoderHead和TransNeXt中的聚合感知注意力改进C2f.(百度云视频-20240106更新说明)
-
ultralytics/cfg/models/yolo-detr/yolov8-detr-SDI.yaml
使用RT-DETR中的TransformerDecoderHead和U-NetV2中的 Semantics and Detail Infusion Module对yolov8中的feature fusion进行改进.
-
ultralytics/cfg/models/yolo-detr/yolov8-detr-goldyolo.yaml
利用RT-DETR中的TransformerDecoderHead和华为2023最新GOLD-YOLO中的Gatherand-Distribute进行改进特征融合模块.
-
ultralytics/cfg/models/yolo-detr/yolov8-detr-goldyolo-asf.yaml
利用RT-DETR中的TransformerDecoderHead和华为2023最新GOLD-YOLO中的Gatherand-Distribute和ASF-YOLO中的Attentional Scale Sequence Fusion进行改进特征融合模块.
-
ultralytics/cfg/models/yolo-detr/yolov8-detr-C2f-DCNV4.yaml
使用DCNV4改进C2f.
-
ultralytics/cfg/models/yolo-detr/yolov8-detr-HSFPN.yaml
利用RT-DETR中的TransformerDecoderHead和使用MFDS-DETR中的HS-FPN改进YOLOV8中的PAN.
-
ultralytics/cfg/models/yolo-detr/yolov8-detr-HSPAN.yaml
利用RT-DETR中的TransformerDecoderHead和对MFDS-DETR中的HS-FPN进行二次创新后得到HSPAN改进YOLOV8中的PAN.
-
ultralytics/cfg/models/yolo-detr/yolov8-detr-Dysample.yaml
使用ICCV2023 DySample改进yolov8-detr neck中的上采样.
-
ultralytics/cfg/models/yolo-detr/yolov8-detr-CARAFE.yaml
使用ICCV2019 CARAFE改进yolov8-detr neck中的上采样.
-
ultralytics/cfg/models/yolo-detr/yolov8-detr-HWD.yaml
使用Haar wavelet downsampling改进yolov8-detr neck的下采样.
-
ultralytics/cfg/models/yolo-detr/yolov8-detr-ASF-Dynamic.yaml
使用ICCV2023 DySample改进ASF-YOLO中的Attentional Scale Sequence Fusion的上采样模块得到Dynamic Sample Attentional Scale Sequence Fusion改进yolov8-detr中的neck.
-
ultralytics/cfg/models/yolo-detr/yolov8-detr-C2f-SWC.yaml
使用shift-wise conv改进yolov8-detr中的C2f.
-
ultralytics/cfg/models/yolo-detr/yolov8-detr-iRMB-DRB.yaml
使用UniRepLKNet中的DilatedReparamBlock对EMO ICCV2023中的iRMB进行二次创新来改进yolov8-detr中的C2f.
-
ultralytics/cfg/models/yolo-detr/yolov8-detr-iRMB-SWC.yaml
使用shift-wise conv对EMO ICCV2023中的iRMB进行二次创新来改进yolov8-detr中的C2f.
以Yolov5为基准模型的改进方案
-
ultralytics/cfg/models/yolo-detr/yolov5-detr.yaml
使用RT-DETR中的TransformerDecoderHead改进yolov5.
-
ultralytics/cfg/models/yolo-detr/yolov5-detr-DWR.yaml
使用RT-DETR中的TransformerDecoderHead和DWRSeg中的Dilation-wise Residual(DWR)模块改进yolov5.
-
ultralytics/cfg/models/yolo-detr/yolov5-detr-fasternet.yaml
使用RT-DETR中的TransformerDecoderHead和FasterNet CVPR2023改进yolov5.(支持替换其他主干,请看百度云视频-替换主干示例教程)
-
ultralytics/cfg/models/yolo-detr/yolov5-detr-AIFI-LPE.yaml
使用RT-DETR中的TransformerDecoderHead和LearnedPositionalEncoding改进yolov5.(详细介绍请看百度云视频-20231119更新说明)
-
ultralytics/cfg/models/yolo-detr/yolov5-detr-C3-DCNV2.yaml
使用RT-DETR中的TransformerDecoderHead和可变形卷积DCNV2改进yolov5.
-
ultralytics/cfg/models/yolo-detr/yolov5-detr-C3-DCNV3.yaml
使用RT-DETR中的TransformerDecoderHead和可变形卷积DCNV3 CVPR2023改进yolov5.(安装教程请看百度云视频-20231119更新说明)
-
ultralytics/cfg/models/yolo-detr/yolov5-detr-C3-DCNV2-Dynamic.yaml
使用RT-DETR中的TransformerDecoderHead和自研可变形卷积DCNV2-Dynamic改进yolov5.(详细介绍请看百度云视频-MPCA与DCNV2_Dynamic的说明)
-
ultralytics/cfg/models/yolo-detr/yolov5-detr-C3-Ortho.yaml(详细介绍请看百度云视频-20231119更新说明)
使用RT-DETR中的TransformerDecoderHead和OrthoNets中的正交通道注意力改进yolov5.
-
ultralytics/cfg/models/yolo-detr/yolov5-detr-attention.yaml
添加注意力到基于RTDETR-Head中的yolov5中.(手把手教程请看百度云视频-手把手添加注意力教程)
-
ultralytics/cfg/models/yolo-detr/yolov5-detr-p2.yaml
添加小目标检测头P2到TransformerDecoderHead中.
-
ultralytics/cfg/models/yolo-detr/yolov5-detr-C3-DySnake.yaml
DySnakeConv与C3融合.
-
ultralytics/cfg/models/yolo-detr/yolov5-detr-C3-Faster.yaml
使用RT-DETR中的TransformerDecoderHead和FasterNet CVPR2023中的Faster-Block改进yolov5.
-
ultralytics/cfg/models/yolo-detr/yolov5-detr-C3-Faster-Rep.yaml
使用RT-DETR中的TransformerDecoderHead和FasterNet CVPR2023中与RepVGG CVPR2021中的RepConv二次创新后的Faster-Block-Rep改进yolov5.
-
ultralytics/cfg/models/yolo-detr/yolov5-detr-C3-Faster-EMA.yaml
使用RT-DETR中的TransformerDecoderHead和FasterNet CVPR2023中与EMA ICASSP2023二次创新后的Faster-Block-EMA的Faster-Block-EMA改进yolov5.
-
ultralytics/cfg/models/yolo-detr/yolov5-detr-C3-Faster-Rep-EMA.yaml
使用RT-DETR中的TransformerDecoderHead和FasterNet CVPR2023中与RepVGG CVPR2021中的RepConv、EMA ICASSP2023二次创新后的Faster-Block改进yolov5.
-
ultralytics/cfg/models/yolo-detr/yolov5-detr-C3-AKConv.yaml
使用RT-DETR中的TransformerDecoderHead和AKConv 2023改进yolov5.
-
ultralytics/cfg/models/yolo-detr/yolov5-detr-C3-RFAConv.yaml
使用RT-DETR中的TransformerDecoderHead和RFAConv 2023改进yolov5.
-
ultralytics/cfg/models/yolo-detr/yolov5-detr-C3-RFAConv.yaml
使用RT-DETR中的TransformerDecoderHead和RFCAConv 2023改进yolov5.
-
ultralytics/cfg/models/yolo-detr/yolov5-detr-C3-RFAConv.yaml
使用RT-DETR中的TransformerDecoderHead和RFCBAMConv 2023改进yolov5.
-
ultralytics/cfg/models/yolo-detr/yolov5-detr-C3-Conv3XC.yaml
使用RT-DETR中的TransformerDecoderHead和Swift Parameter-free Attention Network中的Conv3XC改进yolov5.
-
ultralytics/cfg/models/yolo-detr/yolov5-detr-C3-SPAB.yaml
使用RT-DETR中的TransformerDecoderHead和Swift Parameter-free Attention Network中的SPAB改进yolov5.
-
ultralytics/cfg/models/yolo-detr/yolov5-detr-C3-DRB.yaml
使用RT-DETR中的TransformerDecoderHead和UniRepLKNet中的DilatedReparamBlock改进改进yolov5.
-
ultralytics/cfg/models/yolo-detr/yolov5-detr-C3-UniRepLKNetBlock.yaml
使用RT-DETR中的TransformerDecoderHead和UniRepLKNet中的UniRepLKNetBlock改进改进yolov5.
-
ultralytics/cfg/models/yolo-detr/yolov5-detr-DWR-DRB.yaml
使用UniRepLKNet中的DilatedReparamBlock对DWRSeg中的Dilation-wise Residual(DWR)进行二次创新改进yolov5.
-
ultralytics/cfg/models/yolo-detr/yolov5-detr-C3-DBB.yaml
使用RT-DETR中的TransformerDecoderHead和DiverseBranchBlock CVPR2021改进yolov5.
-
ultralytics/cfg/models/yolo-detr/yolov5-detr-CSP-EDLAN.yaml
使用RT-DETR中的TransformerDecoderHead和DualConv打造CSP Efficient Dual Layer Aggregation Networks改进yolov5.
-
ultralytics/cfg/models/yolo-detr/yolov5-detr-ASF.yaml
使用RT-DETR中的TransformerDecoderHead和ASF-YOLO中的Attentional Scale Sequence Fusion改进yolov5.
-
ultralytics/cfg/models/yolo-detr/yolov5-detr-ASF-P2.yaml
在ultralytics/cfg/models/yolo-detr/yolov5-detr-ASF.yaml的基础上进行二次创新,引入P2检测层并对网络结构进行优化.
-
ultralytics/cfg/models/yolo-detr/yolov5-detr-slimneck.yaml
使用RT-DETR中的TransformerDecoderHead和SlimNeck中VoVGSCSP\VoVGSCSPC和GSConv改进yolov5的neck.
-
ultralytics/cfg/models/yolo-detr/yolov5-detr-slimneck-asf.yaml
在ultralytics/cfg/models/yolo-detr/yolov5-detr-slimneck.yaml使用ASF-YOLO中的Attentional Scale Sequence Fusion进行二次创新.
-
ultralytics/cfg/models/yolo-detr/yolov5-detr-C3-AggregatedAtt.yaml
使用RT-DETR中的TransformerDecoderHead和TransNeXt中的聚合感知注意力改进C3.(百度云视频-20240106更新说明)
-
ultralytics/cfg/models/yolo-detr/yolov5-detr-SDI.yaml
使用RT-DETR中的TransformerDecoderHead和U-NetV2中的 Semantics and Detail Infusion Module对yolov5中的feature fusion进行改进.
-
ultralytics/cfg/models/yolo-detr/yolov5-detr-goldyolo.yaml
利用RT-DETR中的TransformerDecoderHead和华为2023最新GOLD-YOLO中的Gatherand-Distribute进行改进特征融合模块.
-
ultralytics/cfg/models/yolo-detr/yolov5-detr-goldyolo-asf.yaml
利用RT-DETR中的TransformerDecoderHead和华为2023最新GOLD-YOLO中的Gatherand-Distribute和ASF-YOLO中的Attentional Scale Sequence Fusion进行改进特征融合模块.
-
ultralytics/cfg/models/yolo-detr/yolov5-detr-C3-DCNV4.yaml
使用DCNV4改进C3.
-
ultralytics/cfg/models/yolo-detr/yolov5-detr-HSFPN.yaml
利用RT-DETR中的TransformerDecoderHead和使用MFDS-DETR中的HS-FPN改进YOLOV5中的PAN.
-
ultralytics/cfg/models/yolo-detr/yolov5-detr-HSPAN.yaml
利用RT-DETR中的TransformerDecoderHead和对MFDS-DETR中的HS-FPN进行二次创新后得到HSPAN改进YOLOV5中的PAN.
-
ultralytics/cfg/models/yolo-detr/yolov8-detr-Dysample.yaml
使用ICCV2023 DySample改进yolov8-detr neck中的上采样.
-
ultralytics/cfg/models/yolo-detr/yolov8-detr-CARAFE.yaml
使用ICCV2019 CARAFE改进yolov8-detr neck中的上采样.
-
ultralytics/cfg/models/yolo-detr/yolov8-detr-HWD.yaml
使用Haar wavelet downsampling改进yolov8-detr neck的下采样.
-
ultralytics/cfg/models/yolo-detr/yolov5-detr-ASF-Dynamic.yaml
使用ICCV2023 DySample改进ASF-YOLO中的Attentional Scale Sequence Fusion的上采样模块得到Dynamic Sample Attentional Scale Sequence Fusion改进yolov5-detr中的neck.
-
ultralytics/cfg/models/yolo-detr/yolov5-detr-SWC.yaml
使用shift-wise conv改进yolov5-detr中的C3.
-
ultralytics/cfg/models/yolo-detr/yolov5-detr-iRMB-DRB.yaml
使用UniRepLKNet中的DilatedReparamBlock对EMO ICCV2023中的iRMB进行二次创新来改进yolov5-detr中的C2f.
-
ultralytics/cfg/models/yolo-detr/yolov5-detr-iRMB-SWC.yaml
使用shift-wise conv对EMO ICCV2023中的iRMB进行二次创新来改进yolov5-detr中的C2f.
更新公告
-
20231105-rtdetr-v1.0
- 初版项目发布.
-
20231109-rtdetr-v1.1
- 修复断点训练不能正常使用的bug.
- 优化get_FPS.py中的模型导入方法.
- 增加以yolov5和yolov8为基准模型更换为RTDETR的Head,后续也会提供yolov5-detr,yolov8-detr相关的改进.
- 新增百度云视频-20231109更新说明视频和替换主干说明视频.
- 新增GhostHGNetV2,RepHGNetV2,详细请看使用教程中的RT-DETR改进方案.
- 新增使用DWRSeg中的Dilation-wise Residual(DWR)模块,加强从网络高层的可扩展感受野中提取特征,详细请看使用教程中的RT-DETR改进方案.
-
20231119-rtdetr-v1.2
- 增加DCNV2,DCNV3,DCNV2-Dynamic,并以RTDETR-R18,RTDETR-R50,YOLOV5-Detr,YOLOV8-Detr多个基准模型进行改进,详细请看使用教程中的RT-DETR改进方案.
- 使用CVPR2022-OrthoNets中的正交通道注意力改进resnet18-backbone中的BasicBlock,resnet50-backbone中的BottleNeck,yolov8-C2f,yolov5-C3,详细请看使用教程中的RT-DETR改进方案.
- 使用LearnedPositionalEncoding改进AIFI中的位置编码信息生成,详细请看使用教程中的RT-DETR改进方案.
- 增加EMO模型中的iRMB模块,并使用(EfficientViT-CVPR2023)中的CascadedAttention对其二次创新得到iRMB_Cascaded,详细请看使用教程中的RT-DETR改进方案.
- 百度云视频增加1119更新说明和手把手添加注意力机制视频教学.
- 更新使用教程.
-
20231126-rtdetr-v1.3
- 支持IoU,GIoU,DIoU,CIoU,EIoU,SIoU.
- 支持MPDIoU,Inner-IoU,Inner-MPDIoU.
- 支持Normalized Gaussian Wasserstein Distance.
- 支持小目标检测层P2.
- 支持DySnakeConv.
- 新增Pconv,PConv-Rep(二次创新)优化rtdetr-r18与rtdetr-r50.
- 新增Faster-Block,Faster-Block-Rep(二次创新),Faster-Block-EMA(二次创新),Faster-Block-Rep-EMA(二次创新)优化rtdetr-r18、rtdetr-r50、yolov5-detr、yolov8-retr.
- 更新使用教程.
- 百度云视频增加1126更新说明.
-
20231202-rtdetr-v1.4
- 支持AKConv(具有任意采样形状和任意数目参数的卷积核).
- 支持RFAConv,RFCAConv,RFCBAMConv(感受野注意力卷积).
- 支持UniRepLKNet(大核CNNRepLK正统续作).
- 使用CVPR2022 DAttention改进AIFI.
- 更新使用教程.
- 百度云视频增加1202更新说明.
- 解决训练过程中由于指标出现的nan问题导致best.pt没办法正常保存.
-
20231210-rtdetr-v1.5
- 支持来自Swift Parameter-free Attention Network中的重参数化Conv3XC模块.
- 支持UniRepLKNet中的DilatedReparamBlock.
- 支持UniRepLKNet中的DilatedReparamBlock对DWRSeg中的Dilation-wise Residual(DWR)模块进行二次创新的DWR_DRB.
- 使用ICCV2023 FLatten Transformer中的FocusedLinearAttention改进AIFI.
- 更新使用教程.
- 百度云视频增加1210更新说明.
-
20231214-rtdetr-v1.6
- 支持DiverseBranchBlock.
- 利用DualConv打造CSP Efficient Dual Layer Aggregation Networks(仅支持yolov5-detr和yolov8-detr).
- 使用Swift Parameter-free Attention Network中的重参数化Conv3XC和DiverseBranchBlock改进RepC3.
- 支持最新的ASF-YOLO中的Attentional Scale Sequence Fusion.
- 更新使用教程.
- 百度云视频增加1214更新说明.
-
20231223-rtdetr-v1.7
- 增加rtdetr-r18-asf-p2.yaml,使用ASF-YOLO中的Attentional Scale Sequence Fusion与Small Object Detection Head进行二次创新.
- 新增rtdetr-slimneck.yaml和rtdetr-slimneck-ASF.yaml.
- 新增yolov8-detr-slimneck.yaml,yolov8-detr-slimneck-asf.yaml.
- 新增yolov5-detr-slimneck.yaml,yolov5-detr-slimneck-asf.yaml.
- 修正热力图计算中预处理.
- 更新使用教程.
- 百度云视频增加1223更新说明.
-
20240106-rtdetr-v1.8
- 新增Shape-IoU,Inner-Shape-IoU.
- 新增支持TransNeXt主干和TransNeXt中的聚焦感知注意力机制.
- 新增U-NetV2中的Semantics and Detail Infusion Module对RTDETR的CCFM进行创新.
- ASF系列支持attention_add.
- 更新使用教程.
- 百度云视频增加20240106更新说明.
-
20240113-rtdetr-v1.9
- 支持Wise-IoU(v1,v2,v3)系列(IoU,WIoU,EIoU,GIoU,DIoU,CIoU,SIoU,MPDIoU,ShapeIoU).
- 支持Inner-Wise-IoU(v1,v2,v3)系列(IoU,WIoU,EIoU,GIoU,DIoU,CIoU,SIoU,MPDIoU,ShapeIoU).
- 支持SlideLoss,EMASlideLoss(利用Exponential Moving Average优化mean iou,可当自研创新模块).
- 使用华为2023最新GOLD-YOLO中的Gatherand-Distribute进行改进特征融合模块.
- 使用ASF-YOLO中Attentional Scale Sequence Fusion与GOLD-YOLO中的Gatherand-Distribute进行二次创新结合.
- 修正rtdetr-r34中检测头参数错误的问题,增加rtdetr-r34,rtdetr-r50-m的预训练权重.
- 更新使用教程.
- 百度云视频增加20240113更新说明.
-
20240120-rtdetr-v1.10
- 新增DCNV4.
- 使用LITv2中具有提取高低频信息的高效注意力对AIFI进行二次改进.
- 使用MFDS-DETR中的HS-FPN改进RTDETR中的CCFM和YOLOV5-DETR、YOLOV8-DETR中的Neck.
- 对MFDS-DETR中的HS-FPN进行二次创新后得到HSPAN改进RTDETR中的CCFM和YOLOV5-DETR、YOLOV8-DETR中的Neck.
- 修复没有使用wiou时候断点续寻的bug.
- 修复plot_result.py画结果图中乱码的问题.
- 更新使用教程.
- 百度云视频增加20240120更新说明.
-
20240128-rtdetr-v1.11
- 增加CARAFE轻量化上采样算子.
- 增加DySample(ICCV2023)动态上采样算子.
- 增加Haar wavelet downsampling下采样算子.
- 增加Focaler-IoU,Focaler-GIoU,Focaler-DIoU,Focaler-CIoU,Focaler-EIoU,Focaler-SIoU,Focaler-Shape-IoU,Focaler-MPDIoU.
- 增加Focaler-Wise-IoU(v1,v2,v3)(IoU,WIoU,EIoU,GIoU,DIoU,CIoU,SIoU,MPDIoU,ShapeIoU).
- 使用DySample(ICCV2023)动态上采样算子对ASF-YOLO中的Attentional Scale Sequence Fusion进行二次创新.
- 更新使用教程.
- 百度云视频增加20240128更新说明.
-
20240206-rtdetr-v1.12
- 新增Shift-ConvNets相关改进内容.(rtdetr-SWC.yaml,rtdetr-R50-SWC.yaml,yolov8-detr-C2f-SWC.yaml,yolov5-detr-C3-SWC.yaml)
- 使用UniRepLKNet中的DilatedReparamBlock对EMO中的iRMB进行二次创新.
- 使用Shift-ConvNets中的具有移位操作的卷积对EMO中的iRMB进行二次创新.
- 更新使用教程.
- 百度云视频增加20240206更新说明.
这篇关于RTDETR改进系列指南的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!