From 7c96f8cc02a6a14318087442babf96ddda567f8b Mon Sep 17 00:00:00 2001 From: yejianwu Date: Thu, 1 Mar 2018 18:16:41 +0800 Subject: [PATCH] update config example --- config | 17 ----------------- example.yaml | 25 +++++++++++++++++++++++++ 2 files changed, 25 insertions(+), 17 deletions(-) delete mode 100644 config create mode 100644 example.yaml diff --git a/config b/config deleted file mode 100644 index f47bb827..00000000 --- a/config +++ /dev/null @@ -1,17 +0,0 @@ -; [models] -; DIRECTORIES = tools/mobile_net/64/,tools/mobile_net/128/ - -[configs] -; If exists 'models' section, this section will be global variable to sub models -; Otherwise this is the model's config and must be put in the same dirs with model.pb -TF_INPUT_NODE=input -TF_OUTPUT_NODE=output -MODEL_TAG=mobile_net -INPUT_SHAPE=1,64,64,3 -OUTPUT_SHAPE=1,64,64,2 -RUNTIME=gpu -TARGET_ABI=armeabi-v7a -LIMIT_OPENCL_KERNEL_TIME=0 -DSP_MODE=0 -BENCHMARK_FLAG=0 -EMBED_MODEL_DATA=1 diff --git a/example.yaml b/example.yaml new file mode 100644 index 00000000..4c4b3d17 --- /dev/null +++ b/example.yaml @@ -0,0 +1,25 @@ +# example.yaml +# Each yaml file describes a exported library (could be named [target_abi]/libmace-${filename}.a), +# which can contains more than one models +target_abi: armeabi-v7a # arm64-v8a +embed_model_data: 1 +models: + preview_net: + tf_model_file_path: path/to/model64.pb + tf_input_node: input + tf_output_node: softmax/Reshape_1 + input_shape: 1,64,64,3 + output_shape: 1,64,64,2 + runtime: gpu + limit_opencl_kernel_time: 0 + dsp_mode: 0 + capture_net: + tf_model_file_path: path/to/model256.pb + tf_input_node: input_node + tf_output_node: softmax/Reshape_1 + input_shape: 1,256,256,3 + output_shape: 1,256,256,2 + runtime: gpu + limit_opencl_kernel_time: 1 + dsp_mode: 0 + -- GitLab