fix shitu readme doc

PaddlePaddle · Nov 8, 2022 · e577101 · e577101
1 parent b1e4b4f
commit e577101
Showing 1 changed file with 12 additions and 8 deletions.
diff --git a/docs/zh_CN/models/PP-ShiTu/README.md b/docs/zh_CN/models/PP-ShiTu/README.md
@@ -87,18 +87,22 @@ PP-ShiTuV2 是基于 PP-ShiTuV1 改进的一个实用轻量级通用图像识别
   python3.7 setup.py install
   ```
 
-- 然后执行以下命令下载并解压好demo数据，最后执行一行命令体验图像识别
+- 然后执行以下命令下载并解压好demo数据，最后执行命令体验索引库构建和图像识别
 
   ```shell
   # 下载并解压demo数据
   wget -nc https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/rec/data/drink_dataset_v2.0.tar && tar -xf drink_dataset_v2.0.tar
 
-  # 执行识别命令
-  paddleclas \
-  --model_name=PP-ShiTuV2 \
-  --infer_imgs=./drink_dataset_v2.0/test_images/100.jpeg \
-  --index_dir=./drink_dataset_v2.0/index/ \
-  --data_file=./drink_dataset_v2.0/gallery/drink_label.txt
+  # 在命令行中构建索引库
+  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 \
+  -o IndexProcess.data_file=./drink_dataset_v2.0/gallery/drink_label.txt
+
+  # 执行图像识别命令
+  paddleclas --model_name=PP-ShiTuV2 --predict_type=shitu \
+  -o Global.infer_imgs='./drink_dataset_v2.0/test_images/100.jpeg' \
+  -o IndexProcess.index_dir='./drink_dataset_v2.0/index'
   ```
 
 ## 3 模块介绍与训练
@@ -167,7 +171,7 @@ cd deploy/models
 wget -nc 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 -nc 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.tar
+wget -nc 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
 ```
 
 ### 4.2 测试数据准备