25 lines
538 B
YAML
25 lines
538 B
YAML
name: To ONNX from Tensorflow SavedModel
|
|
inputs:
|
|
- {name: Model, type: TensorflowSavedModel}
|
|
outputs:
|
|
- {name: Model, type: OnnxModel}
|
|
metadata:
|
|
annotations:
|
|
author: Alexey Volkov <alexey.volkov@ark-kun.com>
|
|
implementation:
|
|
container:
|
|
image: tensorflow/tensorflow:2.3.0
|
|
command:
|
|
- sh
|
|
- -exc
|
|
- python3 -m pip install tf2onnx==1.6.3 && "$0" "$@"
|
|
- python3
|
|
- -m
|
|
- tf2onnx.convert
|
|
- --saved-model
|
|
- {inputPath: Model}
|
|
- --output
|
|
- {outputPath: Model}
|
|
- --fold_const
|
|
- --verbose
|