Fix missing IMX500 export decorator (#18823)
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
This commit is contained in:
parent
4a0dffe26c
commit
5bdcf9116d
1 changed files with 3 additions and 0 deletions
|
|
@ -1174,6 +1174,7 @@ class Exporter:
|
|||
yaml_save(export_path / "metadata.yaml", self.metadata)
|
||||
return export_path, None
|
||||
|
||||
@try_export
|
||||
def export_imx(self, prefix=colorstr("IMX:")):
|
||||
"""YOLO IMX export."""
|
||||
gptq = False
|
||||
|
|
@ -1191,6 +1192,8 @@ class Exporter:
|
|||
import onnx
|
||||
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:
|
||||
out = subprocess.run(
|
||||
["java", "--version"], check=True, capture_output=True
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue