Skip to content

Commit

Permalink
Add quantized InceptionV3 on CPU and DSP
Browse files Browse the repository at this point in the history
  • Loading branch information
lee-bin committed Jan 30, 2019
1 parent 8dcf1b0 commit d4713f0
Show file tree
Hide file tree
Showing 3 changed files with 64 additions and 1 deletion.
4 changes: 3 additions & 1 deletion inception-v3/README.md
Expand Up @@ -5,4 +5,6 @@ This folder contains Inception V3 MACE model deployment configurations.

Introduction
---
The original pretrained model is [here](https://github.com/tensorflow/models/tree/master/research/slim#pre-trained-models)
The original pretrained model is [here](https://github.com/tensorflow/models/tree/master/research/slim#pre-trained-models).

The quantization-aware trained model is [here](https://github.com/tensorflow/tensorflow/blob/master/tensorflow/lite/g3doc/models.md#image-classification-quantized-models) (Inception_V3_quant).
33 changes: 33 additions & 0 deletions inception-v3/inception-v3-quantize-retrain-dsp.yml
@@ -0,0 +1,33 @@
library_name: inception-v3-quantize-retrain-dsp
target_abis: [armeabi-v7a]
target_socs: [sdm660]
model_graph_format: file
model_data_format: file
models:
inception_v3_quantize_retrain_dsp:
platform: tensorflow
model_file_path: https://cnbj1.fds.api.xiaomi.com/mace/miai-models/inception-v3/inception_v3_quant_frozen.pb
model_sha256_checksum: 8247ba8c753128da5086a8a5eb85191e2b6bde59d4ba7435b9d07ae053ee3e2e
subgraphs:
- input_tensors:
- input
input_shapes:
- 1,299,299,3
input_ranges:
- -1.0,1.0
output_tensors:
- output
output_shapes:
- 1,1001
check_tensors:
- InceptionV3/Logits/Conv2d_1c_1x1/BiasAdd:0
check_shapes:
- 1,1,1,1001
validation_inputs_data:
- https://cnbj1.fds.api.xiaomi.com/mace/inputs/keyboard_299x299.bin
runtime: dsp
limit_opencl_kernel_time: 0
nnlib_graph_mode: 0
obfuscate: 0
winograd: 0
quantize: 1
28 changes: 28 additions & 0 deletions inception-v3/inception-v3-quantize-retrain.yml
@@ -0,0 +1,28 @@
library_name: inception-v3-quantize-retrain
target_abis: [armeabi-v7a]
model_graph_format: file
model_data_format: file
models:
inception_v3_quantize_retrain:
platform: tensorflow
model_file_path: https://cnbj1.fds.api.xiaomi.com/mace/miai-models/inception-v3/inception_v3_quant_frozen.pb
model_sha256_checksum: 8247ba8c753128da5086a8a5eb85191e2b6bde59d4ba7435b9d07ae053ee3e2e
subgraphs:
- input_tensors:
- input
input_shapes:
- 1,299,299,3
input_ranges:
- -1.0,1.0
output_tensors:
- output
output_shapes:
- 1,1001
validation_inputs_data:
- https://cnbj1.fds.api.xiaomi.com/mace/inputs/keyboard_299x299.bin
runtime: cpu
limit_opencl_kernel_time: 0
nnlib_graph_mode: 0
obfuscate: 0
winograd: 0
quantize: 1

0 comments on commit d4713f0

Please sign in to comment.