diff --git a/docs/en/integrations/sony-imx500.md b/docs/en/integrations/sony-imx500.md index 5dccbc5585..dcd5f16726 100644 --- a/docs/en/integrations/sony-imx500.md +++ b/docs/en/integrations/sony-imx500.md @@ -41,7 +41,7 @@ Export an Ultralytics YOLOv8 model to IMX500 format and run inference with the e !!! note - Here we perform inference just to make sure the model works as expected. However, for deployment and inference on the Raspberry Pi AI Camera, please jump to [Using IMX500 Export in Deployment](#using-imx500-export-in-deployment) section. + IMX export is currently only supported for the YOLOv8n model. Here we perform inference just to make sure the model works as expected. However, for deployment and inference on the Raspberry Pi AI Camera, please jump to [Using IMX500 Export in Deployment](#using-imx500-export-in-deployment) section. !!! example diff --git a/ultralytics/engine/exporter.py b/ultralytics/engine/exporter.py index 9e83dc3c03..796f3fb208 100644 --- a/ultralytics/engine/exporter.py +++ b/ultralytics/engine/exporter.py @@ -1137,7 +1137,7 @@ class Exporter: if getattr(self.model, "end2end", False): raise ValueError("IMX export is not supported for end2end models.") if "C2f" not in self.model.__str__(): - raise ValueError("IMX export is only supported for YOLOv8 detection models") + raise ValueError("IMX export is only supported for YOLOv8n detection models") check_requirements(("model-compression-toolkit==2.1.1", "sony-custom-layers==0.2.0", "tensorflow==2.12.0")) check_requirements("imx500-converter[pt]==3.14.3") # Separate requirements for imx500-converter