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

Fix missing IMX500 export decorator (#18823)

Browse source 
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
This commit is contained in:
Glenn Jocher 2025-01-22 13:58:28 +01:00 • committed by GitHub
parent 4a0dffe26c
commit 5bdcf9116d
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 3 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

3
ultralytics/engine/exporter.py
View file

@ -1174,6 +1174,7 @@ class Exporter:
yaml_save(export_path / "metadata.yaml", self.metadata) yaml_save(export_path / "metadata.yaml", self.metadata)
return export_path, None return export_path, None
@try_export
def export_imx(self, prefix=colorstr("IMX:")): def export_imx(self, prefix=colorstr("IMX:")):
"""YOLO IMX export.""" """YOLO IMX export."""
gptq = False gptq = False
@ -1191,6 +1192,8 @@ class Exporter:
import onnx import onnx
from sony_custom_layers.pytorch.object_detection.nms import multiclass_nms 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: try:
out = subprocess.run( out = subprocess.run(
["java", "--version"], check=True, capture_output=True ["java", "--version"], check=True, capture_output=True