26 lines
1.0 KiB
YAML
26 lines
1.0 KiB
YAML
name: Convert Keras HDF5 model to Tensorflow JS GraphModel
|
|
inputs:
|
|
- {name: Model, type: KerasModelHdf5}
|
|
outputs:
|
|
- {name: Model, type: TensorflowJSGraphModel}
|
|
metadata:
|
|
annotations:
|
|
author: Alexey Volkov <alexey.volkov@ark-kun.com>
|
|
canonical_location: 'https://raw.githubusercontent.com/Ark-kun/pipeline_components/master/components/_converters/TensorflowJSGraphModel/from_KerasModelHdf5/component.yaml'
|
|
implementation:
|
|
container:
|
|
image: tensorflow/tensorflow:2.3.0
|
|
command:
|
|
- sh
|
|
- -exc
|
|
- |
|
|
# Manually installing prerequisites so that tensorflowjs does not re-install tensorflow-cpu on top of tensorflow. See https://github.com/tensorflow/tfjs/issues/3953
|
|
python3 -m pip install --quiet 'h5py>=2.8.0' 'numpy>=1.16.4,<1.19.0' 'six>=1.12.0' 'tensorflow-hub==0.7.0' 'PyInquirer==1.0.3'
|
|
python3 -m pip install --quiet tensorflowjs==2.4.0 --no-dependencies
|
|
"$0" "$*"
|
|
- tensorflowjs_converter
|
|
- --input_format=keras
|
|
- --output_format=tfjs_graph_model
|
|
- inputPath: Model
|
|
- outputPath: Model
|