From 5bdcf9116d6284f24b0ad303cc921eccecff8c24 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Wed, 22 Jan 2025 13:58:28 +0100
Subject: [PATCH] Fix missing IMX500 export decorator (#18823)

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
---
 ultralytics/engine/exporter.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/ultralytics/engine/exporter.py b/ultralytics/engine/exporter.py
index 840fa2a2..cc18e179 100644
--- a/ultralytics/engine/exporter.py
+++ b/ultralytics/engine/exporter.py
@@ -1174,6 +1174,7 @@ class Exporter:
         yaml_save(export_path / "metadata.yaml", self.metadata)
         return export_path, None
 
+    @try_export
     def export_imx(self, prefix=colorstr("IMX:")):
         """YOLO IMX export."""
         gptq = False
@@ -1191,6 +1192,8 @@ class Exporter:
         import onnx
         from sony_custom_layers.pytorch.object_detection.nms import multiclass_nms
 
+        LOGGER.info(f"\n{prefix} starting export with model_compression_toolkit {mct.__version__}...")
+
         try:
             out = subprocess.run(
                 ["java", "--version"], check=True, capture_output=True