branching for explain mode

openvinotoolkit · Jan 20, 2025 · 7a6da22 · 7a6da22
1 parent c3ce361
commit 7a6da22
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/otx/algo/detection/rtdetr.py b/src/otx/algo/detection/rtdetr.py
@@ -128,6 +128,9 @@ def _customize_inputs(
                 )
             targets.append({"boxes": scaled_bboxes, "labels": ll})
 
+        if self.explain_mode:
+            return {"entity": entity}
+
         return {
             "images": entity.images,
             "targets": targets,