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

Fix IMX onnx model path (#18813)

Browse source
This commit is contained in:
Laughing 2025-01-22 19:05:38 +08:00 • committed by GitHub
parent 5306a8cc1f
commit e17c1cdfe0
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
ultralytics/engine/exporter.py
View file

@ -1286,7 +1286,7 @@ class Exporter:
f = Path(str(self.file).replace(self.file.suffix, "_imx_model")) f = Path(str(self.file).replace(self.file.suffix, "_imx_model"))
f.mkdir(exist_ok=True) 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( mct.exporter.pytorch_export_model(
model=quant_model, save_model_path=onnx_model, repr_dataset=representative_dataset_gen model=quant_model, save_model_path=onnx_model, repr_dataset=representative_dataset_gen
) )