From c8db2dd1ef48bdd14f476faa392fe83604099020 Mon Sep 17 00:00:00 2001 From: WJJ1995 Date: Mon, 17 Oct 2022 11:31:00 +0800 Subject: [PATCH] Add YOLOv5Cls API (#382) * add yolov5cls * fixed bugs * fixed bugs * fixed preprocess bug * add yolov5cls readme * deal with comments * Add YOLOv5Cls Note * add yolov5cls test * update yolov5cls api * update yolov5cls api Co-authored-by: Jason --- docs/api_docs/python/image_classification.md | 8 ++++++++ examples/vision/classification/yolov5cls/cpp/README.md | 2 +- examples/vision/classification/yolov5cls/python/README.md | 2 +- 3 files changed, 10 insertions(+), 2 deletions(-) mode change 100644 => 100755 docs/api_docs/python/image_classification.md mode change 100644 => 100755 examples/vision/classification/yolov5cls/cpp/README.md mode change 100644 => 100755 examples/vision/classification/yolov5cls/python/README.md diff --git a/docs/api_docs/python/image_classification.md b/docs/api_docs/python/image_classification.md old mode 100644 new mode 100755 index 66405d63d..46760eec3 --- a/docs/api_docs/python/image_classification.md +++ b/docs/api_docs/python/image_classification.md @@ -7,3 +7,11 @@ :members: :inherited-members: ``` + +## fastdeploy.vision.classification.YOLOv5Cls + +```{eval-rst} +.. autoclass:: fastdeploy.vision.classification.YOLOv5Cls + :members: + :inherited-members: +``` diff --git a/examples/vision/classification/yolov5cls/cpp/README.md b/examples/vision/classification/yolov5cls/cpp/README.md old mode 100644 new mode 100755 index 7778874d3..5611e82a5 --- a/examples/vision/classification/yolov5cls/cpp/README.md +++ b/examples/vision/classification/yolov5cls/cpp/README.md @@ -19,7 +19,7 @@ make -j #下载官方转换好的yolov5模型文件和测试图片 wget https://bj.bcebos.com/paddlehub/fastdeploy/yolov5n-cls.onnx -wget hhttps://gitee.com/paddlepaddle/PaddleClas/raw/release/2.4/deploy/images/ImageNet/ILSVRC2012_val_00000010.jpeg +wget https://gitee.com/paddlepaddle/PaddleClas/raw/release/2.4/deploy/images/ImageNet/ILSVRC2012_val_00000010.jpeg # CPU推理 diff --git a/examples/vision/classification/yolov5cls/python/README.md b/examples/vision/classification/yolov5cls/python/README.md old mode 100644 new mode 100755 index a59e894f6..4b024b298 --- a/examples/vision/classification/yolov5cls/python/README.md +++ b/examples/vision/classification/yolov5cls/python/README.md @@ -12,7 +12,7 @@ git clone https://github.com/PaddlePaddle/FastDeploy.git cd examples/vision/classification/yolov5cls/python/ -#下载yolov5cls模型文件和测试图片 +#下载 YOLOv5Cls 模型文件和测试图片 wget https://bj.bcebos.com/paddlehub/fastdeploy/yolov5n-cls.onnx wget https://gitee.com/paddlepaddle/PaddleClas/raw/release/2.4/deploy/images/ImageNet/ILSVRC2012_val_00000010.jpeg