From e17c1cdfe0425be84b2c0e28e117276a1bb6c038 Mon Sep 17 00:00:00 2001 From: Laughing <61612323+Laughing-q@users.noreply.github.com> Date: Wed, 22 Jan 2025 19:05:38 +0800 Subject: [PATCH] Fix IMX onnx model path (#18813) --- ultralytics/engine/exporter.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ultralytics/engine/exporter.py b/ultralytics/engine/exporter.py index 867e319d..840fa2a2 100644 --- a/ultralytics/engine/exporter.py +++ b/ultralytics/engine/exporter.py @@ -1286,7 +1286,7 @@ class Exporter: f = Path(str(self.file).replace(self.file.suffix, "_imx_model")) f.mkdir(exist_ok=True) - onnx_model = f / Path(str(self.file).replace(self.file.suffix, "_imx.onnx")) # js dir + onnx_model = f / Path(str(self.file.name).replace(self.file.suffix, "_imx.onnx")) # js dir mct.exporter.pytorch_export_model( model=quant_model, save_model_path=onnx_model, repr_dataset=representative_dataset_gen )