Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
PaddleSlim
提交
637e165f
P
PaddleSlim
项目概览
PaddlePaddle
/
PaddleSlim
1 年多 前同步成功
通知
51
Star
1434
Fork
344
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
53
列表
看板
标记
里程碑
合并请求
16
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
PaddleSlim
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
53
Issue
53
列表
看板
标记
里程碑
合并请求
16
合并请求
16
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
未验证
提交
637e165f
编写于
5月 17, 2022
作者:
C
Chang Xu
提交者:
GitHub
5月 17, 2022
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
add ACT mobilenetv1 demo (#1109)
上级
79a7b164
变更
4
隐藏空白更改
内联
并排
Showing
4 changed file
with
149 addition
and
1 deletion
+149
-1
demo/auto_compression/image_classification/README.md
demo/auto_compression/image_classification/README.md
+99
-0
demo/auto_compression/image_classification/configs/mobilev1.yaml
...to_compression/image_classification/configs/mobilev1.yaml
+29
-0
demo/auto_compression/image_classification/run.py
demo/auto_compression/image_classification/run.py
+1
-1
demo/auto_compression/image_classification/run.sh
demo/auto_compression/image_classification/run.sh
+20
-0
未找到文件。
demo/auto_compression/image_classification/README.md
0 → 100644
浏览文件 @
637e165f
# 图像分类模型自动压缩示例
目录:
-
[
1. 简介
](
#1简介
)
-
[
2. Benchmark
](
#2Benchmark
)
-
[
3. 自动压缩流程
](
#自动压缩流程
)
-
[
3.1 准备环境
](
#31-准备准备
)
-
[
3.2 准备数据集
](
#32-准备数据集
)
-
[
3.3 准备预测模型
](
#33-准备预测模型
)
-
[
3.4 自动压缩并产出模型
](
#34-自动压缩并产出模型
)
-
[
4. 预测部署
](
#4预测部署
)
-
[
5. FAQ
](
5FAQ
)
## 1. 简介
本示例将以图像分类模型MobileNetV1为例,介绍如何使用PaddleClas中Inference部署模型进行自动压缩。本示例使用的自动压缩策略为量化训练和蒸馏。
## 2. Benchmark
-
MobileNetV1模型
| 模型 | 策略 | Top-1 Acc | 耗时(ms) threads=4 |
|:------:|:------:|:------:|:------:|
| MobileNetV1 | Base模型 | 70.90 | 39.041 |
| MobileNetV1 | 量化+蒸馏 | 70.49 | 29.238|
-
测试环境:
`SDM710 2*A75(2.2GHz) 6*A55(1.7GHz)`
## 3. 自动压缩流程
#### 3.1 准备环境
-
python >= 3.6
-
PaddlePaddle >= 2.2 (可从
[
Paddle官网
](
https://www.paddlepaddle.org.cn/install/quick?docurl=/documentation/docs/zh/install/pip/linux-pip.html
)
下载安装)
-
PaddleSlim >= 2.3 或者适当develop版本
安装paddlepaddle:
```
shell
# CPU
pip
install
paddlepaddle
# GPU
pip
install
paddlepaddle-gpu
```
安装paddleslim:
```
shell
pip
install
paddleslim
```
#### 3.2 准备数据集
本案例默认以ImageNet1k数据进行自动压缩实验,如数据集为非ImageNet1k格式数据, 请参考
[
PaddleClas数据准备文档
](
https://github.com/PaddlePaddle/PaddleClas/blob/release/2.3/docs/zh_CN/data_preparation/classification_dataset.md
)
。
#### 3.3 准备预测模型
预测模型的格式为:
`model.pdmodel`
和
`model.pdiparams`
两个,带
`pdmodel`
的是模型文件,带
`pdiparams`
后缀的是权重文件。
注:其他像
`__model__`
和
`__params__`
分别对应
`model.pdmodel`
和
`model.pdiparams`
文件。
可在
[
PaddleClas预训练模型库
](
https://github.com/PaddlePaddle/PaddleClas/blob/release/2.3/docs/zh_CN/algorithm_introduction/ImageNet_models.md
)
中直接获取Inference模型,具体可参考下方获取MobileNetV1模型示例:
```
shell
wget https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/inference/MobileNetV1_infer.tar
tar
-zxvf
MobileNetV1_infer.tar
```
也可根据
[
PaddleClas文档
](
https://github.com/PaddlePaddle/PaddleClas/blob/release/2.3/docs/zh_CN/inference_deployment/export_model.md
)
导出Inference模型。
#### 3.4 自动压缩并产出模型
蒸馏量化自动压缩示例通过run.py脚本启动,会使用接口
```paddleslim.auto_compression.AutoCompression```
对模型进行量化训练和蒸馏。配置config文件中模型路径、数据集路径、蒸馏、量化和训练等部分的参数,配置完成后便可开始自动压缩。
```
shell
# 单卡启动
python run.py
\
--model_dir
=
'MobileNetV1_infer'
\
--model_filename
=
'inference.pdmodel'
\
--params_filename
=
'inference.pdiparams'
\
--save_dir
=
'./save_quant_mobilev1/'
\
--batch_size
=
128
\
--config_path
=
'./configs/mobilev1.yaml'
\
--data_dir
=
'ILSVRC2012'
# 多卡启动
python
-m
paddle.distributed.launch run.py
\
--model_dir
=
'MobileNetV1_infer'
\
--model_filename
=
'inference.pdmodel'
\
--params_filename
=
'inference.pdiparams'
\
--save_dir
=
'./save_quant_mobilev1/'
\
--batch_size
=
128
\
--config_path
=
'./configs/mobilev1.yaml'
\
--data_dir
=
'ILSVRC2012'
```
## 4.预测部署
-
[
Paddle Inference Python部署
](
https://github.com/PaddlePaddle/PaddleSeg/blob/release/2.5/docs/deployment/inference/python_inference.md
)
-
[
Paddle Inference C++部署
](
https://github.com/PaddlePaddle/PaddleSeg/blob/release/2.5/docs/deployment/inference/cpp_inference.md
)
-
[
Paddle Lite部署
](
https://github.com/PaddlePaddle/PaddleSeg/blob/release/2.5/docs/deployment/lite/lite.md
)
## 5.FAQ
\ No newline at end of file
demo/auto_compression/image_classification/configs/mobilev1.yaml
0 → 100644
浏览文件 @
637e165f
Distillation
:
distill_lambda
:
1.0
distill_loss
:
l2_loss
distill_node_pair
:
-
teacher_softmax_0.tmp_0
-
softmax_0.tmp_0
merge_feed
:
true
teacher_model_dir
:
MobileNetV1_infer
teacher_model_filename
:
inference.pdmodel
teacher_params_filename
:
inference.pdiparams
Quantization
:
activation_bits
:
8
is_full_quantize
:
false
activation_quantize_type
:
range_abs_max
weight_quantize_type
:
abs_max
not_quant_pattern
:
-
skip_quant
quantize_op_types
:
-
conv2d
-
depthwise_conv2d
weight_bits
:
8
TrainConfig
:
epochs
:
1
eval_iter
:
500
learning_rate
:
0.004
optimizer
:
Momentum
optim_args
:
weight_decay
:
0.00003
origin_metric
:
0.70898
\ No newline at end of file
demo/auto_compression/image_classification/run.py
浏览文件 @
637e165f
import
os
import
sys
sys
.
path
[
0
]
=
os
.
path
.
join
(
os
.
path
.
dirname
(
"__file__"
),
os
.
path
.
pardir
)
sys
.
path
[
0
]
=
os
.
path
.
join
(
os
.
path
.
dirname
(
"__file__"
),
os
.
path
.
pardir
,
os
.
path
.
pardir
)
import
argparse
import
functools
from
functools
import
partial
...
...
demo/auto_compression/image_classification/run.sh
0 → 100644
浏览文件 @
637e165f
# 单卡启动
python run.py
\
--model_dir
=
'MobileNetV1_infer'
\
--model_filename
=
'inference.pdmodel'
\
--params_filename
=
'inference.pdiparams'
\
--save_dir
=
'./save_quant_mobilev1/'
\
--batch_size
=
128
\
--config_path
=
'./configs/mobilev1.yaml'
\
--data_dir
=
'ILSVRC2012'
# 多卡启动
# python -m paddle.distributed.launch run.py \
# --model_dir='MobileNetV1_infer' \
# --model_filename='inference.pdmodel' \
# --params_filename='inference.pdiparams' \
# --save_dir='./save_quant_mobilev1/' \
# --batch_size=128 \
# --config_path='./configs/mobilev1.yaml'\
# --data_dir='/workspace/dataset/ILSVRC2012/'
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录