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

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

        1.ultralytics庫(kù)中RT-DETR模型解析

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

1.ultralytics庫(kù)中RT-DETR模型解析

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

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