ONNX export Path to str() (#12177)

* Update export.py

Signed-off-by: Luis Filipe Araujo de Souza <58831491+Doquey@users.noreply.github.com>

* Update export.py

Signed-off-by: Luis Filipe Araujo de Souza <58831491+Doquey@users.noreply.github.com>

* Update export.py

Transformed the f variable into a string on the export onnx. This bug was making it impossible to export any models in .onnx, since it was making the typehint not accept the users input as it is specified in the functions documentation

Signed-off-by: Luis Filipe Araujo de Souza <58831491+Doquey@users.noreply.github.com>

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

---------

Signed-off-by: Luis Filipe Araujo de Souza <58831491+Doquey@users.noreply.github.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
This commit is contained in:
Luis Filipe Araujo de Souzaandpre-commit-ci[bot] authored and GitHub committed 2023-09-29 19:23:05 +02:00
1 parent 6262c7feb4
commit bb9706e7d1
1 file changed
+1 -1
+1 -1
View File
@@ -155,7 +155,7 @@ def export_onnx(model, im, file, opset, dynamic, simplify, prefix=colorstr('ONNX
import onnx
LOGGER.info(f'\n{prefix} starting export with onnx {onnx.__version__}...')
f = file.with_suffix('.onnx')
f = str(file.with_suffix('.onnx'))
output_names = ['output0', 'output1'] if isinstance(model, SegmentationModel) else ['output0']
if dynamic: