Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
PaddleClas
提交
aa82bef9
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看板
未验证
提交
aa82bef9
编写于
11月 02, 2021
作者:
littletomatodonkey
提交者:
GitHub
11月 02, 2021
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
fix config (#1378)
上级
af700756
变更
5
隐藏空白更改
内联
并排
Showing
5 changed file
with
107 addition
and
16 deletion
+107
-16
deploy/configs/build_general.yaml
deploy/configs/build_general.yaml
+36
-0
deploy/configs/build_product.yaml
deploy/configs/build_product.yaml
+4
-4
deploy/configs/inference_general.yaml
deploy/configs/inference_general.yaml
+55
-0
deploy/configs/inference_product.yaml
deploy/configs/inference_product.yaml
+4
-4
docs/zh_CN/quick_start/quick_start_recognition.md
docs/zh_CN/quick_start/quick_start_recognition.md
+8
-8
未找到文件。
deploy/configs/build_general.yaml
0 → 100644
浏览文件 @
aa82bef9
Global
:
rec_inference_model_dir
:
"
./models/general_PPLCNet_x2_5_lite_v1.0_infer"
batch_size
:
32
use_gpu
:
True
enable_mkldnn
:
True
cpu_num_threads
:
10
enable_benchmark
:
True
use_fp16
:
False
ir_optim
:
True
use_tensorrt
:
False
gpu_mem
:
8000
enable_profile
:
False
RecPreProcess
:
transform_ops
:
-
ResizeImage
:
size
:
224
-
NormalizeImage
:
scale
:
0.00392157
mean
:
[
0.485
,
0.456
,
0.406
]
std
:
[
0.229
,
0.224
,
0.225
]
order
:
'
'
-
ToCHWImage
:
RecPostProcess
:
null
# indexing engine config
IndexProcess
:
index_method
:
"
HNSW32"
# supported: HNSW32, IVF, Flat
image_root
:
"
./drink_dataset_v1.0/gallery/"
index_dir
:
"
./drink_dataset_v1.0/index"
data_file
:
"
./drink_dataset_v1.0/gallery/drink_label.txt"
index_operation
:
"
new"
# suported: "append", "remove", "new"
delimiter
:
"
\t
"
dist_type
:
"
IP"
embedding_size
:
512
deploy/configs/build_product.yaml
浏览文件 @
aa82bef9
Global
:
rec_inference_model_dir
:
"
./models/
general_PPLCNet_x2_5_lite
_v1.0_infer"
rec_inference_model_dir
:
"
./models/
product_ResNet50_vd_aliproduct
_v1.0_infer"
batch_size
:
32
use_gpu
:
True
enable_mkldnn
:
True
...
...
@@ -27,9 +27,9 @@ RecPostProcess: null
# indexing engine config
IndexProcess
:
index_method
:
"
HNSW32"
# supported: HNSW32, IVF, Flat
i
mage_root
:
"
./drink_dataset_v1.0/gallery/
"
i
ndex_dir
:
"
./drink_dataset_v1.0/index
"
data_file
:
"
./
drink_dataset_v1.0/gallery/drink_label
.txt"
i
ndex_dir
:
"
./recognition_demo_data_v1.1/gallery_product/index
"
i
mage_root
:
"
./recognition_demo_data_v1.1/gallery_product/
"
data_file
:
"
./
recognition_demo_data_v1.1/gallery_product/data_file
.txt"
index_operation
:
"
new"
# suported: "append", "remove", "new"
delimiter
:
"
\t
"
dist_type
:
"
IP"
...
...
deploy/configs/inference_general.yaml
0 → 100644
浏览文件 @
aa82bef9
Global
:
infer_imgs
:
"
./drink_dataset_v1.0/test_images/nongfu_spring.jpeg"
det_inference_model_dir
:
"
./models/picodet_PPLCNet_x2_5_mainbody_lite_v1.0_infer"
rec_inference_model_dir
:
"
./models/general_PPLCNet_x2_5_lite_v1.0_infer"
rec_nms_thresold
:
0.05
batch_size
:
1
image_shape
:
[
3
,
640
,
640
]
threshold
:
0.2
max_det_results
:
5
labe_list
:
-
foreground
# inference engine config
use_gpu
:
True
enable_mkldnn
:
True
cpu_num_threads
:
10
enable_benchmark
:
True
use_fp16
:
False
ir_optim
:
True
use_tensorrt
:
False
gpu_mem
:
8000
enable_profile
:
False
DetPreProcess
:
transform_ops
:
-
DetResize
:
interp
:
2
keep_ratio
:
false
target_size
:
[
640
,
640
]
-
DetNormalizeImage
:
is_scale
:
true
mean
:
[
0.485
,
0.456
,
0.406
]
std
:
[
0.229
,
0.224
,
0.225
]
-
DetPermute
:
{}
DetPostProcess
:
{}
RecPreProcess
:
transform_ops
:
-
ResizeImage
:
size
:
224
-
NormalizeImage
:
scale
:
0.00392157
mean
:
[
0.485
,
0.456
,
0.406
]
std
:
[
0.229
,
0.224
,
0.225
]
order
:
'
'
-
ToCHWImage
:
RecPostProcess
:
null
# indexing engine config
IndexProcess
:
index_dir
:
"
./drink_dataset_v1.0/index/"
return_k
:
5
score_thres
:
0.5
deploy/configs/inference_product.yaml
浏览文件 @
aa82bef9
Global
:
infer_imgs
:
"
./
drink_dataset_v1.0/test_images/nongfu_spring.jpe
g"
det_inference_model_dir
:
"
./models/p
icodet_PPLCNet_x2_5_mainbody_lite
_v1.0_infer"
rec_inference_model_dir
:
"
./models/
general_PPLCNet_x2_5_lite
_v1.0_infer"
infer_imgs
:
"
./
recognition_demo_data_v1.1/test_product/daoxiangcunjinzhubing_6.jp
g"
det_inference_model_dir
:
"
./models/p
pyolov2_r50vd_dcn_mainbody
_v1.0_infer"
rec_inference_model_dir
:
"
./models/
product_ResNet50_vd_aliproduct
_v1.0_infer"
rec_nms_thresold
:
0.05
batch_size
:
1
...
...
@@ -50,6 +50,6 @@ RecPostProcess: null
# indexing engine config
IndexProcess
:
index_dir
:
"
./
drink_dataset_v1.0/index/
"
index_dir
:
"
./
recognition_demo_data_v1.1/gallery_product/index
"
return_k
:
5
score_thres
:
0.5
docs/zh_CN/quick_start/quick_start_recognition.md
浏览文件 @
aa82bef9
...
...
@@ -38,7 +38,7 @@
| 模型简介 | 推荐场景 | inference模型 | 预测配置文件 | 构建索引库的配置文件 |
| ------------ | ------------- | -------- | ------- | -------- |
| 轻量级通用主体检测模型 | 通用场景 |
[
模型下载链接
](
https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/rec/models/inference/picodet_PPLCNet_x2_5_mainbody_lite_v1.0_infer.tar
)
| - | - |
| 轻量级通用识别模型 | 通用场景 |
[
模型下载链接
](
https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/rec/models/inference/general_PPLCNet_x2_5_lite_v1.0_infer.tar
)
|
[
inference_
product.yaml
](
../../../deploy/configs/inference_product.yaml
)
|
[
build_product.yaml
](
../../../deploy/configs/build_product
.yaml
)
|
| 轻量级通用识别模型 | 通用场景 |
[
模型下载链接
](
https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/rec/models/inference/general_PPLCNet_x2_5_lite_v1.0_infer.tar
)
|
[
inference_
general.yaml
](
../../../deploy/configs/inference_general.yaml
)
|
[
build_general.yaml
](
../../../deploy/configs/build_general
.yaml
)
|
本章节 demo 数据下载地址如下:
[
瓶装饮料数据下载链接
](
https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/rec/data/drink_dataset_v1.0.tar
)
。
...
...
@@ -126,7 +126,7 @@ wget https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/rec/data/drink_da
```
shell
# 下面是使用下载的服务端商品识别模型进行索引库构建
python3.7 python/build_gallery.py
-c
configs/build_
product
.yaml
-o
Global.rec_inference_model_dir
=
./models/product_ResNet50_vd_aliproduct_v1.0_infer
python3.7 python/build_gallery.py
-c
configs/build_
general
.yaml
-o
Global.rec_inference_model_dir
=
./models/product_ResNet50_vd_aliproduct_v1.0_infer
```
<a
name=
"商品识别与检索"
></a>
...
...
@@ -150,9 +150,9 @@ pip install faiss-cpu==1.7.1post2
```
shell
# 使用下面的命令使用GPU进行预测
python3.7 python/predict_system.py
-c
configs/inference_
product
.yaml
python3.7 python/predict_system.py
-c
configs/inference_
general
.yaml
# 使用下面的命令使用CPU进行预测
python3.7 python/predict_system.py
-c
configs/inference_
product
.yaml
-o
Global.use_gpu
=
False
python3.7 python/predict_system.py
-c
configs/inference_
general
.yaml
-o
Global.use_gpu
=
False
```
待检索图像如下所示。
...
...
@@ -184,7 +184,7 @@ python3.7 python/predict_system.py -c configs/inference_product.yaml -o Global.u
```
shell
# 使用下面的命令使用GPU进行预测,如果希望使用CPU预测,可以在命令后面添加 -o Global.use_gpu=False
python3.7 python/predict_system.py
-c
configs/inference_
product
.yaml
-o
Global.infer_imgs
=
"./drink_dataset_v1.0/test_images/"
python3.7 python/predict_system.py
-c
configs/inference_
general
.yaml
-o
Global.infer_imgs
=
"./drink_dataset_v1.0/test_images/"
```
终端中会输出该文件夹内所有图像的识别结果,如下所示。
...
...
@@ -218,7 +218,7 @@ Inference: 150.06470680236816 ms per batch image
```
shell
# 使用下面的命令使用 GPU 进行预测,如果希望使用 CPU 预测,可以在命令后面添加 -o Global.use_gpu=False
python3.7 python/predict_system.py
-c
configs/inference_
product
.yaml
-o
Global.infer_imgs
=
"./drink_dataset_v1.0/test_images/mosilian.jpeg"
python3.7 python/predict_system.py
-c
configs/inference_
general
.yaml
-o
Global.infer_imgs
=
"./drink_dataset_v1.0/test_images/mosilian.jpeg"
```
待检索图像如下所示。
...
...
@@ -251,7 +251,7 @@ python3.7 python/predict_system.py -c configs/inference_product.yaml -o Global.i
使用下面的命令构建
`index`
索引,加速识别后的检索过程。
```
shell
python3.7 python/build_gallery.py
-c
configs/build_
product
.yaml
-o
IndexProcess.data_file
=
"./drink_dataset_v1.0/gallery/drink_label_all.txt"
-o
IndexProcess.index_dir
=
"./drink_dataset_v1.0/index_all"
python3.7 python/build_gallery.py
-c
configs/build_
general
.yaml
-o
IndexProcess.data_file
=
"./drink_dataset_v1.0/gallery/drink_label_all.txt"
-o
IndexProcess.index_dir
=
"./drink_dataset_v1.0/index_all"
```
最终新的索引信息保存在文件夹
`./drink_dataset_v1.0/index_all`
中。
...
...
@@ -264,7 +264,7 @@ python3.7 python/build_gallery.py -c configs/build_product.yaml -o IndexProcess.
```
shell
# 使用下面的命令使用 GPU 进行预测,如果希望使用 CPU 预测,可以在命令后面添加 -o Global.use_gpu=False
python3.7 python/predict_system.py
-c
configs/inference_
product
.yaml
-o
Global.infer_imgs
=
"././drink_dataset_v1.0/test_images/mosilian.jpeg"
-o
IndexProcess.index_dir
=
"./drink_dataset_v1.0/index_all"
python3.7 python/predict_system.py
-c
configs/inference_
general
.yaml
-o
Global.infer_imgs
=
"././drink_dataset_v1.0/test_images/mosilian.jpeg"
-o
IndexProcess.index_dir
=
"./drink_dataset_v1.0/index_all"
```
输出结果如下。
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录