update paddlecv (#5655)

paddlecv/README.md

@@ -6,24 +6,25 @@
 
 | 单模型/串联系统 | 方向 | 模型名称
 |:------:|:-----:|:------:|
-| 单模型 |   图像分类 |  PP-LCNet |
-| 单模型 |   图像分类 |  PP-LCNet v2 |
-| 单模型 |   图像分类 |  PP-HGNet |
-| 单模型 |   目标检测 |  PP-YOLO |
-| 单模型 |   目标检测 |  PP-YOLO v2 |
-| 单模型 |   目标检测 |  PP-YOLOE |
-| 单模型 |   目标检测 |  PP-YOLOE+ |
-| 单模型 |   目标检测 |  PP-PicoDet |
-| 单模型 |   图像分割 |  PP-HumanSeg v2 |
-| 单模型 |   图像分割 |  PP-LiteSeg |
-| 单模型 |   图像分割 |  PP-Matting v1 |
-| 串联系统 |   OCR |  PP-OCR v2 |
-| 串联系统 |   OCR |  PP-OCR v3 |
-| 串联系统 |   OCR |  PP-Structure |
-| 串联系统 |   图像识别 |  PP-ShiTu |
-| 串联系统 |   图像识别 |  PP-ShiTu v2 |
-| 串联系统 |   行人分析 |  PP-Human |
-| 串联系统 |   车辆分析 |  PP-Vehicle |
+| 单模型 |   图像分类 |  [PP-LCNet](configs/single_op/PP-LCNet.yml) |
+| 单模型 |   图像分类 |  [PP-LCNet v2](configs/single_op/PP-LCNetV2.yml) |
+| 单模型 |   图像分类 |  [PP-HGNet](configs/single_op/PP-HGNet.yml) |
+| 单模型 |   目标检测 |  [PP-YOLO](configs/single_op/PP-YOLO.yml) |
+| 单模型 |   目标检测 |  [PP-YOLO v2](configs/single_op/PP-YOLOv2.yml) |
+| 单模型 |   目标检测 |  [PP-YOLOE](configs/single_op/PP-YOLOE.yml) |
+| 单模型 |   目标检测 |  [PP-YOLOE+](configs/single_op/PP-YOLOE+.yml) |
+| 单模型 |   目标检测 |  [PP-PicoDet](configs/single_op/PP-PicoDet.yml) |
+| 单模型 |   图像分割 |  [PP-HumanSeg v2](configs/single_op/PP-HumanSegV2.yml) |
+| 单模型 |   图像分割 |  [PP-LiteSeg](configs/single_op/PP-LiteSeg.yml) |
+| 单模型 |   图像分割 |  [PP-Matting v1](configs/single_op/PP-MattingV1.yml) |
+| 串联系统 |   OCR |  [PP-OCR v2](configs/system/PP-OCRv2.yml) |
+| 串联系统 |   OCR |  [PP-OCR v3](configs/system/PP-OCRv3.yml) |
+| 串联系统 |   OCR |  [PP-Structure](configs/system/PP-Structure.yml) |
+| 串联系统 |   图像识别 |  [PP-ShiTu](configs/system/PP-ShiTu.yml) |
+| 串联系统 |   图像识别 |  [PP-ShiTu v2](configs/system/PP-ShiTuV2.yml) |
+| 串联系统 |   行人分析 |  [PP-Human](configs/system/PP-Human.yml) |
+| 串联系统 |   车辆分析 |  [PP-Vehicle](configs/system/PP-Vehicle.yml) |
+| 串联系统 |   关键点检测 |  [PP-TinyPose](configs/system/PP-TinyPose.yml) |
 
 ## <img src="https://user-images.githubusercontent.com/48054808/157828296-d5eb0ccb-23ea-40f5-9957-29853d7d13a9.png" width="20"/> 文档教程
 

paddlecv/docs/whl.md

@@ -5,7 +5,7 @@
 目前该whl包尚未上传至pypi，因此目前需要通过以下方式安装。
 
 ```shell
-python setup.py bdist_wheel
+bash scripts/build_wheel.sh
 pip install dist/paddlecv-0.1.0-py3-none-any.whl
 ```
 
@@ -15,14 +15,25 @@ pip install dist/paddlecv-0.1.0-py3-none-any.whl
 使用方式如下所示。
 
 * 可以指定task_name或者config_path，来获取所需要预测的系统。当使用`task_name`时，会从PaddleCV项目中获取已经自带的模型或者串联系统，进行预测，而使用`config_path`时，则会加载配置文件，完成模型或者串联系统的初始化。
-s
+
 ```py
 from paddlecv import PaddleCV
 paddlecv = PaddleCV(task_name="PP-OCRv3")
 res = paddlecv("../demo/00056221.jpg")
 ```
 
-* 如果希望查看系统自带的的串联系统列表，可以使用下面的方式。
+PaddleCV参数说明
+
+| 参数名   |  默认值 | 含义 |
+|:------:|:---------:|:---------:|
+|  task_name    |    None |  任务名，和config_path必须指定其一    |
+|  config_path  |    None |  配置文件路径，支持自动下载，和task_name必须指定其一  |
+|  output_dir    |      output |  输出结果保存路径，包含可视化结果和结构化输出          |
+|  run_mode    |      paddle |  预测部署模式，可选项为`'paddle'/'trt_fp32'/'trt_fp16'/'trt_int8'/'mkldnn'/'mkldnn_bf16'`    |
+|  device    |    CPU |  运行设备，可选项为`CPU/GPU/XPU`      |
+
+
+* 如果希望查看系统自带的模型和串联系统列表，可以使用下面的方式。
 
 ```py
 from paddlecv import PaddleCV

paddlecv/ppcv/core/framework.py

@@ -94,8 +94,8 @@ class Executor(object):
     The executor which implements model series pipeline
 
     Args:
-        env_cfg: The enrionment configuration
         model_cfg: The models configuration
+        env_cfg: The enrionment configuration
     """
 
     def __init__(self, model_cfg, env_cfg):

paddlecv/ppcv/model_zoo/model_zoo.py

@@ -31,18 +31,23 @@ __all__ = [
 MODEL_ZOO_FILENAME = 'MODEL_ZOO'
 TASK_DICT = {
     # single model
-    'classification': 'paddlecv://configs/single_op/PP-HGNet',
-    'detection': 'paddlecv://configs/single_op/PP-YOLOE+.yml',
-    'segmentation': 'paddlecv://configs/single_op/PP-LiteSeg.yml',
+    'PP-LCNet': 'paddlecv://configs/single_op/PP-LCNet.yml',
+    'PP-LCNetV2': 'paddlecv://configs/single_op/PP-LCNetV2.yml',
+    'PP-HGNet': 'paddlecv://configs/single_op/PP-HGNet.yml',
+    'PP-YOLOE+': 'paddlecv://configs/single_op/PP-YOLOE+.yml',
+    'PP-YOLOE': 'paddlecv://configs/single_op/PP-YOLOE.yml',
+    'PP-YOLO': 'paddlecv://configs/single_op/PP-YOLO.yml',
+    'PP-YOLOv2': 'paddlecv://configs/single_op/PP-YOLOv2.yml',
+    'PP-PicoDet': 'paddlecv://configs/single_op/PP-PicoDet.yml',
+    'PP-HumanSegV2': 'paddlecv://configs/single_op/PP-HumanSegV2.yml',
+    'PP-LiteSeg': 'paddlecv://configs/single_op/PP-LiteSeg.yml',
+    'PP-MattingV1': 'paddlecv://configs/single_op/PP-MattingV1.yml',
     # system
     'PP-OCRv2': 'paddlecv://configs/system/PP-OCRv2.yml',
     'PP-OCRv3': 'paddlecv://configs/system/PP-OCRv3.yml',
-    'PP-StructureV2': 'paddlecv://configs/system/PP-Structure.yml',
-    'PP-StructureV2-layout-table':
-    'paddlecv://configs/system/PP-Structure-layout-table.yml',
-    'PP-StructureV2-table': 'paddlecv://configs/system/PP-Structure-table.yml',
-    'PP-StructureV2-ser': 'paddlecv://configs/system/PP-Structure-ser.yml',
-    'PP-StructureV2-re': 'paddlecv://configs/system/PP-Structure-re.yml',
+    'PP-Structure': 'paddlecv://configs/system/PP-Structure.yml',
+    'PP-ShiTuV2': 'paddlecv://configs/system/PP-ShiTuV2.yml',
+    'PP-ShiTu': 'paddlecv://configs/system/PP-ShiTu.yml',
     'PP-Human': 'paddlecv://configs/system/PP-Human.yml',
     'PP-Vehicle': 'paddlecv://configs/system/PP-Vehicle.yml',
     'PP-TinyPose': 'paddlecv://configs/system/PP-TinyPose.yml',

paddlecv/tests/test_get_model.py

@@ -26,7 +26,7 @@ import paddle
 import ppcv
 import unittest
 
-TASK_NAME = 'detection'
+TASK_NAME = 'PP-YOLOE'
 MODEL_NAME = 'paddlecv://models/PPLCNet_x1_0_infer/inference.pdiparams'
 
 

paddlecv/tools/predict.py

@@ -52,7 +52,8 @@ def argsparser():
         "--run_mode",
         type=str,
         default='paddle',
-        help="mode of running(paddle/trt_fp32/trt_fp16/trt_int8)")
+        help="mode of running(paddle/trt_fp32/trt_fp16/trt_int8/mkldnn/mkldnn_bf16)"
+    )
     parser.add_argument(
         "--device",
         type=str,