getting_started.md 3.0 KB
Newer Older
W
WuHaobo 已提交
1 2
# 开始使用
---
littletomatodonkey's avatar
littletomatodonkey 已提交
3
请事先参考[安装指南](install.md)配置运行环境,并根据[数据说明](./data.md)文档准备ImageNet1k数据,本章节下面所有的实验均以ImageNet1k数据集为例。
W
WuHaobo 已提交
4

S
fix  
shippingwang 已提交
5
## 一、设置环境变量
W
WuHaobo 已提交
6 7 8 9 10 11 12

**设置PYTHONPATH环境变量:**

```bash
export PYTHONPATH=path_to_PaddleClas:$PYTHONPATH
```

S
fix  
shippingwang 已提交
13
## 二、模型训练与评估
W
WuHaobo 已提交
14

littletomatodonkey's avatar
littletomatodonkey 已提交
15
PaddleClas 提供模型训练与评估脚本:`tools/train.py``tools/eval.py`
W
WuHaobo 已提交
16 17

### 2.1 模型训练
littletomatodonkey's avatar
littletomatodonkey 已提交
18 19

按照如下方式启动模型训练。
W
wangshipeng01 已提交
20

W
WuHaobo 已提交
21 22 23 24 25 26
```bash
# PaddleClas通过launch方式启动多卡多进程训练
# 通过设置FLAGS_selected_gpus 指定GPU运行卡号

python -m paddle.distributed.launch \
    --selected_gpus="0,1,2,3" \
W
WuHaobo 已提交
27
    tools/train.py \
littletomatodonkey's avatar
littletomatodonkey 已提交
28
        -c ./configs/ResNet/ResNet50_vd.yaml
W
WuHaobo 已提交
29 30 31 32 33 34 35 36
```

- 输出日志示例如下:

```
epoch:0    train    step:13    loss:7.9561    top1:0.0156    top5:0.1094    lr:0.100000    elapse:0.193
```

C
cuicheng01 已提交
37
可以通过添加-o参数来更新配置:
W
WuHaobo 已提交
38 39 40 41

```bash
python -m paddle.distributed.launch \
    --selected_gpus="0,1,2,3" \
W
WuHaobo 已提交
42
    tools/train.py \
littletomatodonkey's avatar
littletomatodonkey 已提交
43
        -c ./configs/ResNet/ResNet50_vd.yaml \
S
shippingwang 已提交
44 45
        -o use_mix=1 \
	--vdl_dir=./scalar/
W
WuHaobo 已提交
46 47 48 49 50 51 52 53 54

```

- 输出日志示例如下:

```
epoch:0    train    step:522    loss:1.6330    lr:0.100000    elapse:0.210
```

littletomatodonkey's avatar
littletomatodonkey 已提交
55
也可以直接修改模型对应的配置文件更新配置。具体配置参数参考[配置文档](config.md)
W
fix  
wangshipeng01 已提交
56

S
shippingwang 已提交
57 58 59 60 61 62 63
训练期间可以通过VisualDL实时观察loss变化,启动命令如下:

```bash
visualdl --logdir ./scalar --host <host_IP> --port <port_num>

```

W
fix  
wangshipeng01 已提交
64

littletomatodonkey's avatar
littletomatodonkey 已提交
65
### 2.2 模型微调
W
fix  
wangshipeng01 已提交
66

littletomatodonkey's avatar
littletomatodonkey 已提交
67
* [30分钟玩转PaddleClas](./quick_start.md)中包含大量模型微调的示例,可以参考该章节在特定的数据集上进行模型微调。
W
fix  
wangshipeng01 已提交
68

littletomatodonkey's avatar
littletomatodonkey 已提交
69
### 2.3 模型评估
W
WuHaobo 已提交
70 71

```bash
W
WuHaobo 已提交
72
python tools/eval.py \
littletomatodonkey's avatar
littletomatodonkey 已提交
73
    -c ./configs/eval.yaml \
littletomatodonkey's avatar
littletomatodonkey 已提交
74
    -o ARCHITECTURE.name="ResNet50_vd" \
W
WuHaobo 已提交
75 76
    -o pretrained_model=path_to_pretrained_models
```
littletomatodonkey's avatar
littletomatodonkey 已提交
77
可以更改configs/eval.yaml中的`ARCHITECTURE.name`字段和pretrained_model字段来配置评估模型,也可以通过-o参数更新配置。
S
fix  
shippingwang 已提交
78

littletomatodonkey's avatar
littletomatodonkey 已提交
79 80
**注意:** 加载预训练模型时,需要指定预训练模型的前缀,例如预训练模型参数所在的文件夹为`output/ResNet50_vd/19`,预训练模型参数的名称为`output/ResNet50_vd/19/ppcls.pdparams`,则`pretrained_model`参数需要指定为`output/ResNet50_vd/19/ppcls`,PaddleClas会自动补齐`.pdparams`的后缀。

S
refine  
shippingwang 已提交
81
## 三、模型推理
S
fix  
shippingwang 已提交
82

S
shippingwang 已提交
83
PaddlePaddle提供三种方式进行预测推理,接下来介绍如何用预测引擎进行推理:
C
cuicheng01 已提交
84
首先,对训练好的模型进行转换:
littletomatodonkey's avatar
littletomatodonkey 已提交
85

S
shippingwang 已提交
86 87
```bash
python tools/export_model.py \
88 89 90
    --model=模型名字 \
    --pretrained_model=预训练模型路径 \
    --output_path=预测模型保存路径
S
fix  
shippingwang 已提交
91

S
shippingwang 已提交
92
```
C
cuicheng01 已提交
93
之后,通过预测引擎进行推理:
S
fix  
shippingwang 已提交
94
```bash
W
WuHaobo 已提交
95
python tools/infer/predict.py \
S
shippingwang 已提交
96 97 98 99 100
    -m model文件路径 \
    -p params文件路径 \
    -i 图片路径 \
    --use_gpu=1 \
    --use_tensorrt=True
S
fix  
shippingwang 已提交
101
```
C
cuicheng01 已提交
102
更多使用方法和推理方式请参考[分类预测框架](../extension/paddle_inference.md)