diff --git a/config b/config deleted file mode 100644 index f47bb827ddf176dc36d2302729e96280810d6b20..0000000000000000000000000000000000000000 --- 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 0000000000000000000000000000000000000000..4c4b3d17c3731c1cd69ac1f0a099d70d7908c922 --- /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 +