example.yaml 1.4 KB
Newer Older
Y
yejianwu 已提交
1 2 3
# example.yaml
# Each yaml file describes a exported library (could be named [target_abi]/libmace-${filename}.a), 
# which can contains more than one models
Y
fix run  
yejianwu 已提交
4
# target_soc can get by `adb shell getprop | grep ro.board.platform | cut -d [ -f3 | cut -d ] -f1`
5
target_abis: [armeabi-v7a, arm64-v8a]
Y
yejianwu 已提交
6
target_socs: [MSM8953]
Y
yejianwu 已提交
7 8 9
embed_model_data: 1
models:
  preview_net:
10
    platform: tensorflow
11
    model_file_path: path/to/model64.pb # also support http:// and https://
L
liuqi 已提交
12
    model_sha256_checksum: 05d92625809dc9edd6484882335c48c043397aed450a168d75eb8b538e86881a
L
liuqi 已提交
13 14 15 16
    input_nodes: input_node
    output_nodes: output_node
    input_shapes: 1,64,64,3
    output_shapes: 1,64,64,2
Y
yejianwu 已提交
17 18 19
    runtime: gpu
    limit_opencl_kernel_time: 0
    dsp_mode: 0
20
    obfuscate: 1
L
liuqi 已提交
21
    fast_conv: 0
22 23
    validation_inputs_data:
      - path/to/input_files
Y
yejianwu 已提交
24
  capture_net:
25 26 27
    platform: caffe
    model_file_path: path/to/model.prototxt
    weight_file_path: path/to/weight.caffemodel
L
liuqi 已提交
28 29
    model_sha256_checksum: 05d92625809dc9edd6484882335c48c043397aed450a168d75eb8b538e86881a
    weight_sha256_checksum: 05d92625809dc9edd6484882335c48c043397aed450a168d75eb8b538e86881a
L
liuqi 已提交
30 31 32 33 34 35 36
    input_nodes:
      - input_node0
      - input_node1
    output_nodes:
      - output_node0
      - output_node1
    input_shapes:
37 38
      - 1,256,256,3
      - 1,128,128,3
L
liuqi 已提交
39
    output_shapes:
40 41
      - 1,256,256,2
      - 1,1,1,2
42
    runtime: cpu
Y
yejianwu 已提交
43 44
    limit_opencl_kernel_time: 1
    dsp_mode: 0
45
    obfuscate: 1
L
liuqi 已提交
46
    fast_conv: 0