提交 7c96f8cc 编写于 作者: Y yejianwu

update config example

上级 72b22554
; [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
# 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
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册