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 )