tensorspace-converter icon indicating copy to clipboard operation
tensorspace-converter copied to clipboard

TensorSpace-Converter YAML configuration

Open syt123450 opened this issue 5 years ago • 0 comments

Just a proposal, TensorSpace-Converter load a YAML configuration file and preprocess the model based on YAML configuration.

YAML configuration:

input_model_from: tensorflow
input_model_format: tf_keras
output_layer_names:
 - layer1Name
 - layer2Name
 - layer3Name
input_path: ./PATH/TO/MODEL/xxx.h5
output_path: ./PATH/TO/SAVE/DIR

Converter shell command:

tensorspacejs_converter \
  --configuration_yaml=PATH/TO/YAML/FILE/configuration.yaml

syt123450 avatar Mar 29 '19 02:03 syt123450