diff --git a/docs/en/integrations/sony-imx500.md b/docs/en/integrations/sony-imx500.md index 5dccbc55..dcd5f167 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 9e83dc3c..796f3fb2 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