mirror of https://github.com/alibaba/MNN.git
75 lines
2.3 KiB
Markdown
75 lines
2.3 KiB
Markdown
# Python工具
|
||
Pymnn是MNN的Python版本，其中将部分MNN工具封装成了`MNNTools`，MNNTools模块主要有以下工具：
|
||
- mnn
|
||
- mnnconvert
|
||
- mnnquant
|
||
|
||
### mnn
|
||
mnn工具的功能是列出目前MNNTools支持的所有工具，使用如下：
|
||
```bash
|
||
mnn
|
||
mnn toolsets has following command line tools
|
||
$mnn
|
||
list out mnn commands
|
||
$mnnconvert
|
||
convert other model to mnn model
|
||
$mnnquant
|
||
quantize mnn model
|
||
```
|
||
### mnnconvert
|
||
mnnconvert是对[MNNConvert](convert.md)的Python封装，参数使用可以[参考](convert.html#id2)，示例如下：
|
||
```bash
|
||
mnnconvert -f ONNX --modelFile mobilenetv2-7.onnx --MNNModel mobilenetv2-7.mnn --bizCode mobilenet
|
||
Start to Convert Other Model Format To MNN Model...
|
||
[11:34:53] :40: ONNX Model ir version: 6
|
||
Start to Optimize the MNN Net...
|
||
107 op name is empty or dup, set to Const107
|
||
108 op name is empty or dup, set to BinaryOp108
|
||
109 op name is empty or dup, set to Unsqueeze109
|
||
111 op name is empty or dup, set to Unsqueeze111
|
||
inputTensors : [ input, ]
|
||
outputTensors: [ output, ]
|
||
Converted Success!
|
||
```
|
||
### mnnquant
|
||
mnnquant是对[quantized.out](quant.html#id4)的Python封装，具体用法可以参考[quantized.out](quant.html#id4)，示例如下：
|
||
```bash
|
||
cp /path/to/MNN # using MNN/resource/images as input
|
||
mnnquant shuffle.mnn shuffle_quant.mnn shuffle_quant.json
|
||
[11:48:17] :48: >>> modelFile: shuffle.mnn
|
||
[11:48:17] :49: >>> preTreatConfig: shuffle_quant.json
|
||
[11:48:17] :50: >>> dstFile: shuffle_quant.mnn
|
||
[11:48:17] :77: Calibrate the feature and quantize model...
|
||
[11:48:17] :156: Use feature quantization method: KL
|
||
[11:48:17] :157: Use weight quantization method: MAX_ABS
|
||
[11:48:17] :177: feature_clamp_value: 127
|
||
[11:48:17] :178: weight_clamp_value: 127
|
||
[11:48:17] :111: used image num: 2
|
||
[11:48:17] :668: fake quant weights done.
|
||
ComputeFeatureRange: 100.00 %
|
||
CollectFeatureDistribution: 100.00 %
|
||
[11:48:36] :82: Quantize model done!
|
||
```
|
||
配置文件`shuffle_quant.json`内容如下：
|
||
```json
|
||
{
|
||
"format":"RGB",
|
||
"mean":[
|
||
103.94,
|
||
116.78,
|
||
123.68
|
||
],
|
||
"normal":[
|
||
0.017,
|
||
0.017,
|
||
0.017
|
||
],
|
||
"width":224,
|
||
"height":224,
|
||
"path":"./resource/images/",
|
||
"used_image_num":2,
|
||
"feature_quantize_method":"KL",
|
||
"weight_quantize_method":"MAX_ABS",
|
||
"model":"shuffle.mnn"
|
||
}
|
||
``` |