Add dummy XAI to RTDETR (export mode) & disable strong aug

CHANGELOG.md

@@ -118,6 +118,8 @@ All notable changes to this project will be documented in this file.
   (<https://github.com/openvinotoolkit/training_extensions/pull/4082>)
 - Fix RTMDet Inst Explain Mode
   (<https://github.com/openvinotoolkit/training_extensions/pull/4083>)
+- Fix RTDETR Explain Mode
+  (<https://github.com/openvinotoolkit/training_extensions/pull/4106>)
 
 ## \[v2.1.0\]
 

src/otx/algo/detection/base_models/detection_transformer.py

@@ -5,6 +5,7 @@
 
 from __future__ import annotations
 
+import warnings
 from typing import Any
 
 import numpy as np
@@ -95,16 +96,22 @@ def export(
         explain_mode: bool = False,
     ) -> dict[str, Any] | tuple[list[Any], list[Any], list[Any]]:
         """Exports the model."""
-        if explain_mode:
-            msg = "Explain mode is not supported for DETR models yet."
-            raise NotImplementedError(msg)
-
-        return self.postprocess(
+        results = self.postprocess(
             self._forward_features(batch_inputs),
             [meta["img_shape"] for meta in batch_img_metas],
             deploy_mode=True,
         )
 
+        if explain_mode:
+            # TODO(Eugene): Implement explain mode for DETR model.
+            warnings.warn("Explain mode is not supported for DETR model. Return dummy values.", stacklevel=2)
+            xai_output = {
+                "feature_vector": torch.zeros(1, 1),
+                "saliency_map": torch.zeros(1),
+            }
+            results.update(xai_output)  # type: ignore[union-attr]
+        return results
+
     def postprocess(
         self,
         outputs: dict[str, Tensor],

src/otx/recipe/detection/rtdetr_101.yaml

@@ -54,9 +54,6 @@ overrides:
         - class_path: torchvision.transforms.v2.RandomPhotometricDistort
           init_args:
             p: 0.5
-        - class_path: torchvision.transforms.v2.RandomZoomOut
-          init_args:
-            fill: 0
         - class_path: otx.core.data.transform_libs.torchvision.RandomFlip
           init_args:
             prob: 0.5

src/otx/recipe/detection/rtdetr_18.yaml

@@ -53,9 +53,6 @@ overrides:
         - class_path: torchvision.transforms.v2.RandomPhotometricDistort
           init_args:
             p: 0.5
-        - class_path: torchvision.transforms.v2.RandomZoomOut
-          init_args:
-            fill: 0
         - class_path: otx.core.data.transform_libs.torchvision.RandomFlip
           init_args:
             prob: 0.5

src/otx/recipe/detection/rtdetr_50.yaml

@@ -54,9 +54,6 @@ overrides:
         - class_path: torchvision.transforms.v2.RandomPhotometricDistort
           init_args:
             p: 0.5
-        - class_path: torchvision.transforms.v2.RandomZoomOut
-          init_args:
-            fill: 0
         - class_path: otx.core.data.transform_libs.torchvision.RandomFlip
           init_args:
             prob: 0.5