23 lines
599 B
YAML
23 lines
599 B
YAML
name: Convert Tensorflow SavedModel to Tensorflow Lite model
|
|
inputs:
|
|
- {name: Model, type: TensorflowSavedModel}
|
|
outputs:
|
|
- {name: Model, type: TensorflowLiteModel}
|
|
metadata:
|
|
annotations:
|
|
author: Alexey Volkov <alexey.volkov@ark-kun.com>
|
|
implementation:
|
|
container:
|
|
image: tensorflow/tensorflow:2.3.0
|
|
command:
|
|
- sh
|
|
- -exc
|
|
- |
|
|
model_path="$0"
|
|
output_model_path="$1"
|
|
mkdir -p "$(dirname "$output_model_path")"
|
|
|
|
tflite_convert --saved_model_dir "$model_path" --output_file "$output_model_path"
|
|
- {inputPath: Model}
|
|
- {outputPath: Model}
|