From 93be023a80694d734240278aca0d0c0b308e2d2e Mon Sep 17 00:00:00 2001 From: Bin Lu Date: Wed, 16 Jun 2021 14:36:46 +0800 Subject: [PATCH] Update getting_started_retrieval.md --- docs/zh_CN/tutorials/getting_started_retrieval.md | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/docs/zh_CN/tutorials/getting_started_retrieval.md b/docs/zh_CN/tutorials/getting_started_retrieval.md index 7cdb14f9..256d33fb 100644 --- a/docs/zh_CN/tutorials/getting_started_retrieval.md +++ b/docs/zh_CN/tutorials/getting_started_retrieval.md @@ -168,6 +168,7 @@ python python/build_gallery.py \ + `IndexProcess.index_path`：gallery库index的路径 + `IndexProcess.image_root`：gallery库图片的根目录 + `IndexProcess.data_file`：gallery库图片的文件列表 +执行完上述命令之后，会在`../dataset/flowers102`目录下面生成`index`目录，index目录下面包含3个文件`index.data`, `1index.graph`, `info.json` ### 3.3 推理预测 @@ -186,3 +187,8 @@ python python/predict_rec.py \ + `Global.rec_inference_model_dir`：预测模型文件路径，如 `./inference/` + `Global.use_tensorrt`：是否使用 TesorRT 预测引擎，默认值：`True` + `Global.use_gpu`：是否使用 GPU 预测，默认值：`True` +执行完上述命令之后，会得到输入图片对应的特征信息； 本例子中特征维度为2048；显示如下： +``` +(1, 2048) +[[0.00033124 0.00056205 0.00032261 ... 0.00030939 0.00050748 0.00030271]] +``` -- GitLab