load_model.md 1.2 KB
Newer Older
J
jiangjiajun 已提交
1
# 模型加载
J
jiangjiajun 已提交
2 3 4

PaddleX提供了统一的模型加载接口,支持加载PaddleX保存的模型,并在验证集上进行评估或对测试图片进行预测

J
jiangjiajun 已提交
5 6
## paddlex.load_model
> **加载PaddleX保存的模型**  
J
jiangjiajun 已提交
7 8 9 10 11 12 13 14 15 16 17 18 19

```
paddlex.load_model(model_dir)
```

### 参数

* **model_dir**: 训练过程中保存的模型路径

### 返回值
* **paddlex.cv.models**, 模型类。

### 示例
F
FlyingQianMM 已提交
20 21
> 1. [点击下载](https://bj.bcebos.com/paddlex/models/xiaoduxiong_epoch_12.tar.gz)PaddleX在小度熊分拣数据上训练的MaskRCNN模型
> 2. [点击下载](https://bj.bcebos.com/paddlex/datasets/xiaoduxiong_ins_det.tar.gz)小度熊分拣数据集
J
jiangjiajun 已提交
22 23 24 25

```
import paddlex as pdx

F
FlyingQianMM 已提交
26 27 28
model_dir = './xiaoduxiong_epoch_12'
data_dir = './xiaoduxiong_ins_det/JPEGImages'
ann_file = './xiaoduxiong_ins_det/val.json'
J
jiangjiajun 已提交
29 30 31 32 33

# 加载垃圾分拣模型
model = pdx.load_model(model_dir)

# 预测
F
FlyingQianMM 已提交
34
pred_result = model.predict('./xiaoduxiong_ins_det/JPEGImages/WechatIMG114.jpeg')
J
jiangjiajun 已提交
35 36 37

# 在验证集上进行评估
eval_reader = pdx.cv.datasets.CocoDetection(data_dir=data_dir,
J
Jason 已提交
38
                                            ann_file=ann_file,
J
jiangjiajun 已提交
39 40 41
                                            transforms=model.eval_transforms)
eval_result = model.evaluate(eval_reader, batch_size=1)
```