example.yaml 1.3 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
4
target_abis: [armeabi-v7a, arm64-v8a]
Y
yejianwu 已提交
5
target_socs: [MSM8953]
Y
yejianwu 已提交
6
embed_model_data: 1
Y
yejianwu 已提交
7
vlog_level: 0
Y
yejianwu 已提交
8 9
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
Y
yejianwu 已提交
22
  capture_net:
23 24 25
    platform: caffe
    model_file_path: path/to/model.prototxt
    weight_file_path: path/to/weight.caffemodel
L
liuqi 已提交
26 27
    model_sha256_checksum: 05d92625809dc9edd6484882335c48c043397aed450a168d75eb8b538e86881a
    weight_sha256_checksum: 05d92625809dc9edd6484882335c48c043397aed450a168d75eb8b538e86881a
L
liuqi 已提交
28 29 30 31 32 33 34
    input_nodes:
      - input_node0
      - input_node1
    output_nodes:
      - output_node0
      - output_node1
    input_shapes:
35 36
      - 1,256,256,3
      - 1,128,128,3
L
liuqi 已提交
37
    output_shapes:
38 39
      - 1,256,256,2
      - 1,1,1,2
40
    runtime: cpu
Y
yejianwu 已提交
41 42
    limit_opencl_kernel_time: 1
    dsp_mode: 0
43
    obfuscate: 1
L
liuqi 已提交
44
    fast_conv: 0