最近讀了maskformer以及maskdino的分割頭設(shè)計(jì)，于是想在RT-DETR上做一個分割的改動，所以選擇在ultralytics庫中對RTDETR進(jìn)行改進(jìn)。

本文內(nèi)容簡介：

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

        2. 對ultralytics庫中的RT-DETR模型增加分割頭做實(shí)例分割

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

從yaml文件中可以看出解碼過程是由RTDETRDecoder類實(shí)現(xiàn)的，先看看該類的代碼：

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