Add note that IMX export is only supported for YOLOv8n (#18460)

Signed-off-by: Mohammed Yasin <32206511+Y-T-G@users.noreply.github.com>
pull/18467/head^2
Mohammed Yasin 2 months ago committed by GitHub
parent 114bb0bdb6
commit f932a611ab
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
  1. 2
      docs/en/integrations/sony-imx500.md
  2. 2
      ultralytics/engine/exporter.py

@ -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

@ -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

Loading…
Cancel
Save