Adding RTDETRv2

docs/source/en/_toctree.yml

@@ -683,6 +683,8 @@
         title: ResNet
       - local: model_doc/rt_detr
         title: RT-DETR
+      - local: model_doc/rt_detr_v2
+        title: RtDetrV2
       - local: model_doc/segformer
         title: SegFormer
       - local: model_doc/seggpt

docs/source/en/model_doc/rt_detr_v2.md

@@ -0,0 +1,57 @@
+<!--Copyright 2024 The HuggingFace Team. All rights reserved.
+
+Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with
+the License. You may obtain a copy of the License at
+
+http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on
+an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the
+specific language governing permissions and limitations under the License.
+
+⚠️ Note that this file is in Markdown but contain specific syntax for our doc-builder (similar to MDX) that may not be
+rendered properly in your Markdown viewer.
+
+-->
+
+# RTDetrV2
+
+## Overview
+
+The RTDetrV2 model was proposed in [<INSERT PAPER NAME HERE>](<INSERT PAPER LINK HERE>) by <INSERT AUTHORS HERE>.
+<INSERT SHORT SUMMARY HERE>
+
+The abstract from the paper is the following:
+
+*<INSERT PAPER ABSTRACT HERE>*
+
+Tips:
+
+<INSERT TIPS ABOUT MODEL HERE>
+
+This model was contributed by [INSERT YOUR HF USERNAME HERE](https://huggingface.co/<INSERT YOUR HF USERNAME HERE>).
+The original code can be found [here](<INSERT LINK TO GITHUB REPO HERE>).
+
+
+## RTDetrV2Config
+
+[[autodoc]] RTDetrV2Config
+
+## RTDetrV2ResNetConfig
+
+[[autodoc]] RTDetrV2ResNetConfig
+
+## RTDetrV2Model
+
+[[autodoc]] RTDetrV2Model
+    - forward
+
+## RTDetrV2ForObjectDetection
+
+[[autodoc]] RtDetrV2ForObjectDetection
+    - forward
+
+## RTDetrV2ResNetBackbone
+
+[[autodoc]] RTDetrV2ResNetBackbone
+    - forward

src/transformers/__init__.py

@@ -713,6 +713,7 @@
         "RoFormerTokenizer",
     ],
     "models.rt_detr": ["RTDetrConfig", "RTDetrResNetConfig"],
+    "models.rt_detr_v2": ["RtDetrV2Config"],
     "models.rwkv": ["RwkvConfig"],
     "models.sam": [
         "SamConfig",
@@ -3269,6 +3270,13 @@
             "RTDetrResNetPreTrainedModel",
         ]
     )
+    _import_structure["models.rt_detr_v2"].extend(
+        [
+            "RtDetrV2ForObjectDetection",
+            "RtDetrV2Model",
+            "RtDetrV2PreTrainedModel"
+        ]
+    )
     _import_structure["models.rwkv"].extend(
         [
             "RwkvForCausalLM",
@@ -5616,6 +5624,9 @@
         RTDetrConfig,
         RTDetrResNetConfig,
     )
+    from .models.rt_detr_v2 import (
+        RtDetrV2Config
+    )
     from .models.rwkv import RwkvConfig
     from .models.sam import (
         SamConfig,
@@ -7798,6 +7809,11 @@
             RTDetrResNetBackbone,
             RTDetrResNetPreTrainedModel,
         )
+        from .models.rt_detr_v2 import (
+            RtDetrV2ForObjectDetection,
+            RtDetrV2Model,
+            RtDetrV2PreTrainedModel
+        )
         from .models.rwkv import (
             RwkvForCausalLM,
             RwkvModel,

src/transformers/models/__init__.py

@@ -218,6 +218,7 @@
     roc_bert,
     roformer,
     rt_detr,
+    rt_detr_v2,
     rwkv,
     sam,
     seamless_m4t,

src/transformers/models/auto/configuration_auto.py

@@ -241,6 +241,7 @@
         ("roformer", "RoFormerConfig"),
         ("rt_detr", "RTDetrConfig"),
         ("rt_detr_resnet", "RTDetrResNetConfig"),
+        ("rt_detr_v2", "RtDetrV2Config"),
         ("rwkv", "RwkvConfig"),
         ("sam", "SamConfig"),
         ("seamless_m4t", "SeamlessM4TConfig"),
@@ -557,6 +558,7 @@
         ("roformer", "RoFormer"),
         ("rt_detr", "RT-DETR"),
         ("rt_detr_resnet", "RT-DETR-ResNet"),
+        ("rt_detr_v2", "RtDetrV2"),
         ("rwkv", "RWKV"),
         ("sam", "SAM"),
         ("seamless_m4t", "SeamlessM4T"),

src/transformers/models/auto/modeling_auto.py

@@ -223,6 +223,7 @@
         ("roc_bert", "RoCBertModel"),
         ("roformer", "RoFormerModel"),
         ("rt_detr", "RTDetrModel"),
+        ("rt_detr_v2", "RtDetrV2Model"),
         ("rwkv", "RwkvModel"),
         ("sam", "SamModel"),
         ("seamless_m4t", "SeamlessM4TModel"),
@@ -846,6 +847,7 @@
         ("deta", "DetaForObjectDetection"),
         ("detr", "DetrForObjectDetection"),
         ("rt_detr", "RTDetrForObjectDetection"),
+        ("rt_detr_v2", "RtDetrV2ForObjectDetection"),
         ("table-transformer", "TableTransformerForObjectDetection"),
         ("yolos", "YolosForObjectDetection"),
     ]