HongChuang/ultralytics-ascend
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 3

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

Browse source 
Signed-off-by: Mohammed Yasin <32206511+Y-T-G@users.noreply.github.com>
This commit is contained in:
Mohammed Yasin 2024-12-31 06:09:07 +08:00 • committed by GitHub
parent 114bb0bdb6
commit f932a611ab
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

2
docs/en/integrations/sony-imx500.md
View file

@ -41,7 +41,7 @@ Export an Ultralytics YOLOv8 model to IMX500 format and run inference with the e
!!! note !!! 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 !!! example

2
ultralytics/engine/exporter.py
View file

@ -1137,7 +1137,7 @@ class Exporter:
if getattr(self.model, "end2end", False): if getattr(self.model, "end2end", False):
raise ValueError("IMX export is not supported for end2end models.") raise ValueError("IMX export is not supported for end2end models.")
if "C2f" not in self.model.__str__(): 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(("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 check_requirements("imx500-converter[pt]==3.14.3") # Separate requirements for imx500-converter