From 081c57988827e8d89fb0a18e82c5b74438a66952 Mon Sep 17 00:00:00 2001 From: weishengyu Date: Sat, 19 Jun 2021 00:29:05 +0800 Subject: [PATCH] dbg --- deploy/configs/inference_cls.yaml | 1 - docs/zh_CN/inference.md | 15 +++++++-------- 2 files changed, 7 insertions(+), 9 deletions(-) diff --git a/deploy/configs/inference_cls.yaml b/deploy/configs/inference_cls.yaml index 63808928c..577c77d35 100644 --- a/deploy/configs/inference_cls.yaml +++ b/deploy/configs/inference_cls.yaml @@ -1,4 +1,3 @@ - Global: infer_imgs: "./images/ILSVRC2012_val_00000010.jpeg" inference_model_dir: "./models" diff --git a/docs/zh_CN/inference.md b/docs/zh_CN/inference.md index 7fa25bad6..b989e8035 100644 --- a/docs/zh_CN/inference.md +++ b/docs/zh_CN/inference.md @@ -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 ``` @@ -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 ```