Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
PaddleClas
提交
081c5798
P
PaddleClas
项目概览
PaddlePaddle
/
PaddleClas
大约 1 年 前同步成功
通知
115
Star
4999
Fork
1114
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
19
列表
看板
标记
里程碑
合并请求
6
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
PaddleClas
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
19
Issue
19
列表
看板
标记
里程碑
合并请求
6
合并请求
6
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
提交
081c5798
编写于
6月 19, 2021
作者:
W
weishengyu
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
dbg
上级
55f30c62
变更
2
隐藏空白更改
内联
并排
Showing
2 changed file
with
7 addition
and
9 deletion
+7
-9
deploy/configs/inference_cls.yaml
deploy/configs/inference_cls.yaml
+0
-1
docs/zh_CN/inference.md
docs/zh_CN/inference.md
+7
-8
未找到文件。
deploy/configs/inference_cls.yaml
浏览文件 @
081c5798
Global
:
infer_imgs
:
"
./images/ILSVRC2012_val_00000010.jpeg"
inference_model_dir
:
"
./models"
...
...
docs/zh_CN/inference.md
浏览文件 @
081c5798
...
...
@@ -59,7 +59,7 @@ python3.7 tools/export_model.py -c ppcls/configs/Products/ResNet50_vd_Aliproduct
下载预训练模型:
```
shell script
wget -P ./cls_pretrain/ https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/ResNet50_vd_pretrained.pdparams
wget -P ./cls_pretrain/ https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/
legendary_models/
ResNet50_vd_pretrained.pdparams
```
上述模型是使用ResNet50_vd在ImageNet上训练的模型,使用的配置文件为
`ppcls/configs/ImageNet/ResNet/ResNet50_vd.yaml`
。
...
...
@@ -104,8 +104,7 @@ python3.7 python/predict_det.py -c configs/inference_det.yaml
[](
../images/recognition/product_demo/wangzai.jpg
)
最终输出结果如下:
```
text
[{'class_id': 0, 'score': 0.4762245, 'bbox': array([305.55115, 226.05322, 776.61084, 930.42395],
dtype=float32), 'label_name': 'foreground'}]
[{'class_id': 0, 'score': 0.4762245, 'bbox': array([305.55115, 226.05322, 776.61084, 930.42395], dtype=float32), 'label_name': 'foreground'}]
```
检测的可视化结果如下:
[](
../images/recognition/product_demo/wangzai_det_result.jpg
)
...
...
@@ -119,7 +118,7 @@ python3.7 python/predict_det.py -c configs/inference_det.yaml -o Global.infer_im
如果想使用CPU进行预测,可以将配置文件中use_gpu选项设置为False,或者执行命令:
```
python3
tools/infer/predict_det.py
-o Global.use_gpu=False
python3
.7 python/predict_det.py -c configs/inference_det.yaml
-o Global.use_gpu=False
```
## 三、特征提取模型推理
...
...
@@ -151,7 +150,7 @@ python3.7 python/predict_rec.py -c configs/inference_rec.yaml -o Global.infer_im
如果想使用CPU进行预测,可以将配置文件中use_gpu选项设置为False,或者执行命令:
```
python3
tools/infer/predict_rec.py
-o Global.use_gpu=False
python3
.7 python/predict_rec.py -c configs/inference_rec.yaml
-o Global.use_gpu=False
```
<a
name=
"主体检测、特征提取和向量检索串联"
></a>
...
...
@@ -170,19 +169,19 @@ cd deploy
导出inference模型后,可以使用下面的命令预测:
```
shell script
python3.7 python/predict_cls.py -c configs/inference_
rec
.yaml
python3.7 python/predict_cls.py -c configs/inference_
cls
.yaml
```
如果想要修改图像,可以在configs/inference_det.yaml中,修改infer_imgs的值,或使用-o Global.infer_imgs修改,
例如,要使用
`images/ILSVRC2012_val_00010010.jpeg`
可以运行:
```
shell script
python3.7 python/predict_cls.py -c configs/inference_
rec
.yaml -o Global.infer_imgs=images/ILSVRC2012_val_00010010.jpeg
python3.7 python/predict_cls.py -c configs/inference_
cls
.yaml -o Global.infer_imgs=images/ILSVRC2012_val_00010010.jpeg
```
如果想使用CPU进行预测,可以将配置文件中use_gpu选项设置为False,或者执行命令:
```
python3
tools/infer/predict_rec.py
-o Global.use_gpu=False
python3
.7 python/predict_cls.py -c configs/inference_cls.yaml
-o Global.use_gpu=False
```
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录