diff --git a/docs/user_guide/models/demo_app_models.yml b/docs/user_guide/models/demo_app_models.yml new file mode 100644 index 0000000000000000000000000000000000000000..df130c4b62e1f0278e1cd21675282a71e650c95c --- /dev/null +++ b/docs/user_guide/models/demo_app_models.yml @@ -0,0 +1,39 @@ +# The name of library +library_name: mobilenet +target_abis: [arm64-v8a] +embed_model_data: 1 +# The build mode for model(s). +# 'code' stand for transfer model(s) into cpp code, 'proto' for model(s) in protobuf file(s). +build_type: code +linkshared: 0 +# One yaml config file can contain multi models' config message. +models: + mobilenet_v1: # model tag, which will be used in model loading and must be specific. + platform: tensorflow + # support local path, http:// and https:// + model_file_path: https://cnbj1.fds.api.xiaomi.com/mace/miai-models/mobilenet-v1/mobilenet-v1-1.0.pb + model_sha256_checksum: 71b10f540ece33c49a7b51f5d4095fc9bd78ce46ebf0300487b2ee23d71294e6 + subgraphs: + - input_tensors: input + input_shapes: 1,224,224,3 + output_tensors: MobilenetV1/Predictions/Reshape_1 + output_shapes: 1,1001 + runtime: cpu+gpu + limit_opencl_kernel_time: 0 + nnlib_graph_mode: 0 + obfuscate: 0 + winograd: 0 + mobilenet_v2: + platform: tensorflow + model_file_path: https://cnbj1.fds.api.xiaomi.com/mace/miai-models/mobilenet-v2/mobilenet-v2-1.0.pb + model_sha256_checksum: 369f9a5f38f3c15b4311c1c84c032ce868da9f371b5f78c13d3ea3c537389bb4 + subgraphs: + - input_tensors: input + input_shapes: 1,224,224,3 + output_tensors: MobilenetV2/Predictions/Reshape_1 + output_shapes: 1,1001 + runtime: cpu+gpu + limit_opencl_kernel_time: 0 + nnlib_graph_mode: 0 + obfuscate: 0 + winograd: 0 diff --git a/docs/user_guide/models/demo_app_models_caffe.yml b/docs/user_guide/models/demo_app_models_caffe.yml new file mode 100644 index 0000000000000000000000000000000000000000..e9bf587532aa443526c1906223610340ac47fd03 --- /dev/null +++ b/docs/user_guide/models/demo_app_models_caffe.yml @@ -0,0 +1,24 @@ +# The name of library +library_name: squeezenet-v10 +target_abis: [arm64-v8a] +embed_model_data: 0 +build_type: proto +linkshared: 1 +models: + squeezenet-v10: # model tag, which will be used in model loading and must be specific. + platform: caffe + # support local path, http:// and https:// + model_file_path: https://cnbj1.fds.api.xiaomi.com/mace/miai-models/squeezenet/squeezenet-v1.0.prototxt + weight_file_path: https://cnbj1.fds.api.xiaomi.com/mace/miai-models/squeezenet/squeezenet-v1.0.caffemodel + # sha256_checksum of your model's graph and data files. + # get the sha256_checksum: sha256sum path/to/your/file + model_sha256_checksum: db680cf18bb0387ded9c8e9401b1bbcf5dc09bf704ef1e3d3dbd1937e772cae0 + weight_sha256_checksum: 9ff8035aada1f9ffa880b35252680d971434b141ec9fbacbe88309f0f9a675ce + # define your model's interface + subgraphs: + - input_tensors: data + input_shapes: 1,227,227,3 + output_tensors: prob + output_shapes: 1,1,1,1000 + runtime: cpu+gpu + winograd: 0 diff --git a/docs/user_guide/models/demo_app_models_tf.yml b/docs/user_guide/models/demo_app_models_tf.yml new file mode 100644 index 0000000000000000000000000000000000000000..0600fed2f95c2a8d713d76a85d1f71196d46ad8a --- /dev/null +++ b/docs/user_guide/models/demo_app_models_tf.yml @@ -0,0 +1,23 @@ +# The name of library +library_name: mobilenet +target_abis: [arm64-v8a] +embed_model_data: 0 +build_type: proto +linkshared: 1 +models: + mobilenet_v1: # model tag, which will be used in model loading and must be specific. + platform: tensorflow + # path to your tensorflow model's pb file. Support local path, http:// and https:// + model_file_path: https://cnbj1.fds.api.xiaomi.com/mace/miai-models/mobilenet-v1/mobilenet-v1-1.0.pb + # sha256_checksum of your model's pb file. + # use this command to get the sha256_checksum: sha256sum path/to/your/pb/file + model_sha256_checksum: 71b10f540ece33c49a7b51f5d4095fc9bd78ce46ebf0300487b2ee23d71294e6 + # define your model's interface + subgraphs: + - input_tensors: input + input_shapes: 1,224,224,3 + output_tensors: MobilenetV1/Predictions/Reshape_1 + output_shapes: 1,1001 + # cpu, gpu or cpu+gpu + runtime: cpu+gpu + winograd: 0 \ No newline at end of file