ultralytics库RT-DETR代码解析

最近读了maskformer以及maskdino的分割头设计，于是想在RT-DETR上做一个分割的改动，所以选择在ultralytics库中对RTDETR进行改进。

本文内容简介：

        1.ultralytics库中RT-DETR模型解析

        2. 对ultralytics库中的RT-DETR模型增加分割头做实例分割

---

1.ultralytics库中RT-DETR模型解析

从yaml文件中可以看出解码过程是由RTDETRDecoder类实现的，先看看该类的代码：

class RTDETRDecoder(nn.Module):export = False  # export modedef __init__():super().__init__()def forward(self, x, batch=None):"""Runs the forward pass of the module, returning bounding box and classification scores for the input."""from