diff --git a/paddlecv/README.md b/paddlecv/README.md index ba13e9b150a98068b26f51602d65de4582d79e07..5ba04dce9dea834cedf540e523cf47b54282854a 100644 --- a/paddlecv/README.md +++ b/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) | ## 文档教程 diff --git a/paddlecv/docs/whl.md b/paddlecv/docs/whl.md index 87ec313bc5e5bd1d04d5fa59a015054960f5dfe1..fd3704aa38652dd7e201c611286415c0d93fc0f0 100644 --- a/paddlecv/docs/whl.md +++ b/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 diff --git a/paddlecv/ppcv/core/framework.py b/paddlecv/ppcv/core/framework.py index 9675ab1b0029a7d61efe7c2456124b16192e94a0..d124433134acd86c51b169307649259af4c6b08a 100644 --- a/paddlecv/ppcv/core/framework.py +++ b/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): diff --git a/paddlecv/ppcv/model_zoo/model_zoo.py b/paddlecv/ppcv/model_zoo/model_zoo.py index e5a11803e14fb09d4cf08a7c6ed4694e4011a138..885014e184a6d28f56e7f9abc30b87ffe9792fcb 100644 --- a/paddlecv/ppcv/model_zoo/model_zoo.py +++ b/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', diff --git a/paddlecv/tests/test_get_model.py b/paddlecv/tests/test_get_model.py index a441e9f0169302d2ace781bc136e9c296b05be63..573a6d9085d4819e3880637cd2a6d9867d3a193c 100644 --- a/paddlecv/tests/test_get_model.py +++ b/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' diff --git a/paddlecv/tools/predict.py b/paddlecv/tools/predict.py index 341ebf4fb51d48192354489388bc6111ac9fc501..3d5882d53a0002f2edc75c51bff65b4b93cb9ab9 100644 --- a/paddlecv/tools/predict.py +++ b/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,