PaddleClas/docs/zh_cn/tutorials/getting_started.md

78 lines
1.8 KiB
Markdown
Raw Normal View History

2020-04-10 00:45:02 +08:00
# 开始使用
---
请事先参考[安装指南](install.md)配置运行环境
## 1 设置环境变量
**设置PYTHONPATH环境变量**
```bash
export PYTHONPATH=path_to_PaddleClas:$PYTHONPATH
```
## 2 模型训练与评估
PaddleClas 提供模型训练与评估脚本tools/train.py和tools/eval.py
### 2.1 模型训练
```bash
# PaddleClas通过launch方式启动多卡多进程训练
# 通过设置FLAGS_selected_gpus 指定GPU运行卡号
python -m paddle.distributed.launch \
--selected_gpus="0,1,2,3" \
--log_dir=log_ResNet50 \
train.py \
-c ../configs/ResNet/ResNet50.yaml \
```
- 输出日志示例如下:
```
epoch:0 train step:13 loss:7.9561 top1:0.0156 top5:0.1094 lr:0.100000 elapse:0.193
```
可以通过添加-o参数来更新配置
```bash
python -m paddle.distributed.launch \
--selected_gpus="0,1,2,3" \
--log_dir=log_ResNet50_vd \
train.py \
-c ../configs/ResNet/ResNet50_vd.yaml \
-o use_mix=1 \
```
- 输出日志示例如下:
```
epoch:0 train step:522 loss:1.6330 lr:0.100000 elapse:0.210
```
或是直接修改模型对应的yaml配置文件具体配置参数参考[配置文档](config.md)。
### 2.2 模型评估
```bash
python eval.py \
-c ../configs/eval.yaml \
-o architecture="ResNet50_vd" \
-o pretrained_model=path_to_pretrained_models
```
您可以更改configs/eval.yaml中的architecture字段和pretrained_model字段来配置评估模型或是通过-o参数更新配置。
2020-04-10 09:19:21 +00:00
## 3 模型推理
PaddleClas通过预测引擎进行预测推理
```bash
python tools/predict.py \
-m model文件路径
-p params文件路径
-i 图片路径
--use_tensorrt True
```
更多推理方式和实验请参考[分类预测框架](../extension/paddle_inference.md)