Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
PaddleClas
提交
1649386a
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看板
提交
1649386a
编写于
10月 31, 2022
作者:
悟、
提交者:
zengshao0622
10月 31, 2022
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
cherry pick update quickstart
上级
91389ab9
变更
1
隐藏空白更改
内联
并排
Showing
1 changed file
with
64 addition
and
80 deletion
+64
-80
docs/zh_CN/quick_start/quick_start_recognition.md
docs/zh_CN/quick_start/quick_start_recognition.md
+64
-80
未找到文件。
docs/zh_CN/quick_start/quick_start_recognition.md
浏览文件 @
1649386a
...
@@ -12,7 +12,7 @@
...
@@ -12,7 +12,7 @@
-
[
2. PP-ShiTu PC端 demo 快速体验
](
#2-pp-shitu-pc端-demo-快速体验
)
-
[
2. PP-ShiTu PC端 demo 快速体验
](
#2-pp-shitu-pc端-demo-快速体验
)
-
[
2.1 环境配置
](
#21-环境配置
)
-
[
2.1 环境配置
](
#21-环境配置
)
-
[
2.2 图像识别体验
](
#22-图像识别体验
)
-
[
2.2 图像识别体验
](
#22-图像识别体验
)
-
[
2.2.1 下载、解压
inference 模型与 demo 数据
](
#221-下载解压-inference-模型与
-demo-数据
)
-
[
2.2.1 下载、解压
demo 数据
](
#221-下载解压
-demo-数据
)
-
[
2.2.2 瓶装饮料识别与检索
](
#222-瓶装饮料识别与检索
)
-
[
2.2.2 瓶装饮料识别与检索
](
#222-瓶装饮料识别与检索
)
-
[
2.2.2.1 识别单张图像
](
#2221-识别单张图像
)
-
[
2.2.2.1 识别单张图像
](
#2221-识别单张图像
)
-
[
2.2.2.2 基于文件夹的批量识别
](
#2222-基于文件夹的批量识别
)
-
[
2.2.2.2 基于文件夹的批量识别
](
#2222-基于文件夹的批量识别
)
...
@@ -82,13 +82,17 @@
...
@@ -82,13 +82,17 @@
### 2.1 环境配置
### 2.1 环境配置
*
安装:请先参考文档
[
环境准备
](
../installation.md
)
配置 PaddleClas 运行环境。
*
**[推荐]**
直接 pip 安装:
*
进入
`deploy`
运行目录。本部分所有内容与命令均需要在
`deploy`
目录下运行,可以通过下面的命令进入
`deploy`
目录。
```
bash
pip3
install
paddleclas
```
*
如需使用 PaddleClas develop 分支体验最新功能,或是需要基于 PaddleClas 进行二次开发,请本地构建安装:
```
shell
```
bash
cd
deploy
python3 setup.py
install
```
```
<a
name=
"图像识别体验"
></a>
<a
name=
"图像识别体验"
></a>
...
@@ -125,19 +129,11 @@
...
@@ -125,19 +129,11 @@
<a
name=
"2.2.1"
></a>
<a
name=
"2.2.1"
></a>
#### 2.2.1 下载、解压
inference 模型与
demo 数据
#### 2.2.1 下载、解压 demo 数据
下载 demo 数据集以及轻量级主体检测、识别模型,命令如下。
下载 demo 数据集以及轻量级主体检测、识别模型,命令如下。
```
shell
```
shell
mkdir
models
cd
models
# 下载通用检测 inference 模型并解压
wget https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/rec/models/inference/picodet_PPLCNet_x2_5_mainbody_lite_v1.0_infer.tar
&&
tar
-xf
picodet_PPLCNet_x2_5_mainbody_lite_v1.0_infer.tar
# 下载识别 inference 模型并解压
wget https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/rec/models/inference/PP-ShiTuV2/general_PPLCNetV2_base_pretrained_v1.0_infer.tar
&&
tar
-xf
general_PPLCNetV2_base_pretrained_v1.0_infer.tar
cd
../
# 下载 demo 数据并解压
# 下载 demo 数据并解压
wget https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/rec/data/drink_dataset_v2.0.tar
&&
tar
-xf
drink_dataset_v2.0.tar
wget https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/rec/data/drink_dataset_v2.0.tar
&&
tar
-xf
drink_dataset_v2.0.tar
```
```
...
@@ -155,42 +151,33 @@ wget https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/rec/data/drink_da
...
@@ -155,42 +151,33 @@ wget https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/rec/data/drink_da
其中
`gallery`
文件夹中存放的是用于构建索引库的原始图像,
`index`
表示基于原始图像构建得到的索引库信息,
`test_images`
文件夹中存放的是用于测试识别效果的图像列表。
其中
`gallery`
文件夹中存放的是用于构建索引库的原始图像,
`index`
表示基于原始图像构建得到的索引库信息,
`test_images`
文件夹中存放的是用于测试识别效果的图像列表。
`models`
文件夹下应有如下文件结构:
```
log
├── general_PPLCNetV2_base_pretrained_v1.0_infer
│ ├── inference.pdiparams
│ ├── inference.pdiparams.info
│ └── inference.pdmodel
├── picodet_PPLCNet_x2_5_mainbody_lite_v1.0_infer
│ ├── inference.pdiparams
│ ├── inference.pdiparams.info
│ └── inference.pdmodel
```
**注意**
**注意**
如果使用服务端通用识别模型,Demo 数据需要重新提取特征、够建索引,方式如下:
如果使用服务端通用识别模型,Demo 数据需要重新提取特征、够建索引,方式如下:
**在命令行中构建索引库**
```
shell
```
shell
python3.7 python/build_gallery.py
\
paddleclas
--build_gallery
=
True
--model_name
=
"PP-ShiTuV2"
\
-c
configs/inference_general.yaml
\
-o
IndexProcess.image_root
=
./drink_dataset_v2.0/gallery/
\
-o
Global.rec_inference_model_dir
=
./models/general_PPLCNetV2_base_pretrained_v1.0_infer
-o
IndexProcess.index_dir
=
./drink_dataset_v2.0/index
\
-o
IndexProcess.data_file
=
./drink_dataset_v2.0/gallery/drink_label.txt
```
```
其中参数
`build_gallery(bool)`
控制是否使用索引库构建模式,默认为
`False`
。
<a
name=
"瓶装饮料识别与检索"
></a>
同时可以通过
`-o`
指令更改构建索引库使用的配置,字段说明如下:
#### 2.2.2 瓶装饮料识别与检索
-
IndexProcess.image_root(str): 构建索引库使用的
`gallery`
图像地址。
-
IndexProcess.index_dir(str): 索引库存放地址。
-
IndexProcess.data_file(str): 构建索引库图像的真值文件。
以瓶装饮料识别 demo 为例,展示识别与检索过程(如果希望尝试其他方向的识别与检索效果,在下载解压好对应的 demo 数据与模型之后,替换对应的配置文件即可完成预测)。
注意,此部分使用了
`faiss`
作为检索库,安装方法如下:
```
python
<a
name=
"瓶装饮料识别与检索"
></a>
python3
.
7
-
m
pip
install
faiss
-
cpu
==
1.7
.
1
post2
```
#### 2.2.2 瓶装饮料识别与检索
以瓶装饮料识别 demo 为例,展示识别与检索过程(如果希望尝试其他方向的识别与检索效果,在下载解压好对应的 demo 数据之后,替换命令中对应的文件地址即可)。
若使用时,不能正常引用,则
`uninstall`
之后,重新
`install`
,尤其是 windows 下。
<a
name=
"识别单张图像"
></a>
<a
name=
"识别单张图像"
></a>
...
@@ -202,60 +189,53 @@ python3.7 -m pip install faiss-cpu==1.7.1post2
...
@@ -202,60 +189,53 @@ python3.7 -m pip install faiss-cpu==1.7.1post2
![](
../../images/recognition/drink_data_demo/test_images/100.jpeg
)
![](
../../images/recognition/drink_data_demo/test_images/100.jpeg
)
**在命令行中进行识别和检索**
```
shell
```
shell
# 使用下面的命令使用 GPU 进行预测
paddleclas
--model_name
=
PP-ShiTuV2
--predict_type
=
shitu
\
python3.7 python/predict_system.py
-c
configs/inference_general.yaml
-o
Global.infer_imgs
=
'./drink_dataset_v2.0/test_images/100.jpeg'
\
-o
IndexProcess.index_dir
=
'./drink_dataset_v2.0/index'
# 使用下面的命令使用 CPU 进行预测
python3.7 python/predict_system.py
-c
configs/inference_general.yaml
-o
Global.use_gpu
=
False
```
```
其中参数
`model_name`
为用于检索和识别的模型、
`predict_type`
设置为'shitu'模式。
同时可以通过
`-o`
指令更改检索图像以及索引库,字段说明如下:
-
Global.infer_imgs(str):待检索图像地址。
-
IndexProcess.index_dir(str): 索引库存放地址。
最终输出结果如下。
最终输出结果如下:
```
```
log
[{'bbox': [437, 71, 660, 728], 'rec_docs': '元气森林', 'rec_scores': 0.7740249}, {'bbox': [221, 72, 449, 701], 'rec_docs': '元气森林', 'rec_scores': 0.6950992}, {'bbox': [794, 104, 979, 652], 'rec_docs': '元气森林', 'rec_scores': 0.6305153}], filename: ./drink_dataset_v2.0/test_images/100.jpeg
[{'bbox': [437, 71, 660, 728], 'rec_docs': '元气森林', 'rec_scores': 0.7740249}, {'bbox': [221, 72, 449, 701], 'rec_docs': '元气森林', 'rec_scores': 0.6950992}, {'bbox': [794, 104, 979, 652], 'rec_docs': '元气森林', 'rec_scores': 0.6305153}]
```
```
其中
`bbox`
表示检测出的主体所在位置,
`rec_docs`
表示索引库中与检测框最为相似的类别,
`rec_scores`
表示对应的置信度。
检测的可视化结果默认保存在
`output`
文件夹下,对于本张图像,识别结果可视化如下所示。
![](
../../images/recognition/drink_data_demo/output/100.jpeg
)
其中
`bbox`
表示检测出的主体所在位置,
`rec_docs`
表示索引库中与检测框最为相似的类别,
`rec_scores`
表示对应的置信度。
<a
name=
"基于文件夹的批量识别"
></a>
<a
name=
"基于文件夹的批量识别"
></a>
##### 2.2.2.2 基于文件夹的批量识别
##### 2.2.2.2 基于文件夹的批量识别
如果希望预测文件夹内的图像,可以直接修改
配置文件中的
`Global.infer_imgs`
字段,也可以通过下面的
`-o`
参数修改对应的
配置。
如果希望预测文件夹内的图像,可以直接修改
命令行中的
`-o`
参数对应的
`Global.infer_imgs`
字段
配置。
**在命令行中进行识别和检索**
```
shell
```
shell
# 使用下面的命令使用 GPU 进行预测,如果希望使用 CPU 预测,可以在命令后面添加 -o Global.use_gpu=False
paddleclas
--model_name
=
PP-ShiTuV2
--predict_type
=
shitu
\
python3.7 python/predict_system.py
-c
configs/inference_general.yaml
-o
Global.infer_imgs
=
"./drink_dataset_v2.0/test_images/"
-o
Global.infer_imgs
=
'./drink_dataset_v2.0/test_images'
\
-o
IndexProcess.index_dir
=
'./drink_dataset_v2.0/index'
```
```
终端中会输出该文件夹内所有图像的识别结果,如下所示。
终端中会输出该文件夹内所有图像的识别结果,如下所示。
```
log
```
log
...
...
[{'bbox': [0, 0, 600, 600], 'rec_docs': '红牛-强化型', 'rec_scores': 0.74081033}]
[{'bbox': [0, 0, 600, 600], 'rec_docs': '红牛-强化型', 'rec_scores': 0.74081033}], filename: ./drink_dataset_v2.0/test_images/001.jpeg
Inference: 120.39852142333984 ms per batch image
[{'bbox': [0, 0, 514, 436], 'rec_docs': '康师傅矿物质水', 'rec_scores': 0.6918598}], filename: ./drink_dataset_v2.0/test_images/002.jpeg
[{'bbox': [0, 0, 514, 436], 'rec_docs': '康师傅矿物质水', 'rec_scores': 0.6918598}]
[{'bbox': [138, 40, 573, 1198], 'rec_docs': '乐虎功能饮料', 'rec_scores': 0.68214047}], filename: ./drink_dataset_v2.0/test_images/003.jpeg
Inference: 32.045602798461914 ms per batch image
[{'bbox': [328, 7, 467, 272], 'rec_docs': '脉动', 'rec_scores': 0.60406065}], filename: ./drink_dataset_v2.0/test_images/004.jpeg
[{'bbox': [138, 40, 573, 1198], 'rec_docs': '乐虎功能饮料', 'rec_scores': 0.68214047}]
[{'bbox': [242, 82, 498, 726], 'rec_docs': '味全_每日C', 'rec_scores': 0.5428652}], filename: ./drink_dataset_v2.0/test_images/005.jpeg
Inference: 113.41428756713867 ms per batch image
[{'bbox': [437, 71, 660, 728], 'rec_docs': '元气森林', 'rec_scores': 0.7740249}, {'bbox': [221, 72, 449, 701], 'rec_docs': '元气森林', 'rec_scores': 0.6950992}, {'bbox': [794, 104, 979, 652], 'rec_docs': '元气森林', 'rec_scores': 0.6305153}], filename: ./drink_dataset_v2.0/test_images/100.jpeg
[{'bbox': [328, 7, 467, 272], 'rec_docs': '脉动', 'rec_scores': 0.60406065}]
Inference: 122.04337120056152 ms per batch image
[{'bbox': [242, 82, 498, 726], 'rec_docs': '味全_每日C', 'rec_scores': 0.5428652}]
Inference: 37.95266151428223 ms per batch image
[{'bbox': [437, 71, 660, 728], 'rec_docs': '元气森林', 'rec_scores': 0.7740249}, {'bbox': [221, 72, 449, 701], 'rec_docs': '元气森林', 'rec_scores': 0.6950992}, {'bbox': [794, 104, 979, 652], 'rec_docs': '元气森林', 'rec_scores': 0.6305153}]
...
...
```
```
所有图像的识别结果可视化图像也保存在
`output`
文件夹内。
更多地,可以通过修改
`Global.rec_inference_model_dir`
字段来更改识别 inference 模型的路径,通过修改
`IndexProcess.index_dir`
字段来更改索引库索引的路径。
<a
name=
"未知类别的图像识别体验"
></a>
<a
name=
"未知类别的图像识别体验"
></a>
...
@@ -269,9 +249,11 @@ Inference: 37.95266151428223 ms per batch image
...
@@ -269,9 +249,11 @@ Inference: 37.95266151428223 ms per batch image
执行如下识别命令
执行如下识别命令
**在命令行中进行识别和检索**
```
shell
```
shell
# 使用下面的命令使用 GPU 进行预测,如果希望使用 CPU 预测,可以在命令后面添加 -o Global.use_gpu=False
paddleclas
--model_name
=
PP-ShiTuV2
--predict_type
=
shitu
\
python3.7 python/predict_system.py
-c
configs/inference_general.yaml
-o
Global.infer_imgs
=
"./drink_dataset_v2.0/test_images/mosilian.jpeg"
-o
Global.infer_imgs
=
'./drink_dataset_v2.0/test_images/mosilian.jpeg'
\
-o
IndexProcess.index_dir
=
'./drink_dataset_v2.0/index'
```
```
可以发现输出结果为空
可以发现输出结果为空
...
@@ -296,9 +278,14 @@ python3.7 python/predict_system.py -c configs/inference_general.yaml -o Global.i
...
@@ -296,9 +278,14 @@ python3.7 python/predict_system.py -c configs/inference_general.yaml -o Global.i
使用下面的命令构建新的索引库
`index_all`
。
使用下面的命令构建新的索引库
`index_all`
。
**在命令行中构建索引库**
```
shell
```
shell
python3.7 python/build_gallery.py
-c
configs/inference_general.yaml
-o
IndexProcess.data_file
=
"./drink_dataset_v2.0/gallery/drink_label_all.txt"
-o
IndexProcess.index_dir
=
"./drink_dataset_v2.0/index_all"
paddleclas
--build_gallery
=
True
--model_name
=
"PP-ShiTuV2"
\
-o
IndexProcess.image_root
=
./drink_dataset_v2.0/gallery/
\
-o
IndexProcess.index_dir
=
./drink_dataset_v2.0/index_all
\
-o
IndexProcess.data_file
=
./drink_dataset_v2.0/gallery/drink_label_all.txt
```
```
其中参数
`build_gallery(bool)`
控制是否使用索引库构建模式,默认为
`False`
。
最终构建完毕的新的索引库保存在文件夹
`./drink_dataset_v2.0/index_all`
下。具体
`yaml`
请参考
[
向量检索文档
](
../deployment/PP-ShiTu/vector_search.md
)
。
最终构建完毕的新的索引库保存在文件夹
`./drink_dataset_v2.0/index_all`
下。具体
`yaml`
请参考
[
向量检索文档
](
../deployment/PP-ShiTu/vector_search.md
)
。
...
@@ -308,21 +295,18 @@ python3.7 python/build_gallery.py -c configs/inference_general.yaml -o IndexProc
...
@@ -308,21 +295,18 @@ python3.7 python/build_gallery.py -c configs/inference_general.yaml -o IndexProc
使用新的索引库,重新对
`mosilian.jpeg`
图像进行识别,运行命令如下。
使用新的索引库,重新对
`mosilian.jpeg`
图像进行识别,运行命令如下。
**在命令行中进行识别和检索**
```
shell
```
shell
# 使用下面的命令使用 GPU 进行预测,如果希望使用 CPU 预测,可以在命令后面添加 -o Global.use_gpu=False
paddleclas
--model_name
=
PP-ShiTuV2
--predict_type
=
shitu
\
python3.7 python/predict_system.py
-c
configs/inference_general.yaml
-o
Global.infer_imgs
=
"./drink_dataset_v2.0/test_images/mosilian.jpeg"
-o
IndexProcess.index_dir
=
"./drink_dataset_v2.0/index_all"
-o
Global.infer_imgs
=
'./drink_dataset_v2.0/test_images/mosilian.jpeg'
\
-o
IndexProcess.index_dir
=
'./drink_dataset_v2.0/index_all'
```
```
输出结果如下。
输出结果如下。
```
log
```
log
[{'bbox': [290, 297, 564, 919], 'rec_docs': '光明_莫斯利安', 'rec_scores': 0.59137374}]
[{'bbox': [290, 297, 564, 919], 'rec_docs': '光明_莫斯利安', 'rec_scores': 0.59137374}]
, filename: ./drink_dataset_v2.0/test_images/mosilian.jpeg
```
```
最终识别结果为
`光明_莫斯利安`
,识别正确,识别结果可视化如下所示。
![](
../../images/recognition/drink_data_demo/output/mosilian.jpeg
)
<a
name=
"5"
></a>
<a
name=
"5"
></a>
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录