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 <jiangjiajun@baidu.com>
This commit is contained in:
WJJ1995
2022-10-17 11:31:00 +08:00
committed by GitHub
parent 063e82bb94
commit c8db2dd1ef
3 changed files with 10 additions and 2 deletions

8
docs/api_docs/python/image_classification.md Normal file → Executable file
View File

@@ -7,3 +7,11 @@
:members: :members:
:inherited-members: :inherited-members:
``` ```
## fastdeploy.vision.classification.YOLOv5Cls
```{eval-rst}
.. autoclass:: fastdeploy.vision.classification.YOLOv5Cls
:members:
:inherited-members:
```

2
examples/vision/classification/yolov5cls/cpp/README.md Normal file → Executable file
View File

@@ -19,7 +19,7 @@ make -j
#下载官方转换好的yolov5模型文件和测试图片 #下载官方转换好的yolov5模型文件和测试图片
wget https://bj.bcebos.com/paddlehub/fastdeploy/yolov5n-cls.onnx 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推理 # CPU推理

View File

@@ -12,7 +12,7 @@
git clone https://github.com/PaddlePaddle/FastDeploy.git git clone https://github.com/PaddlePaddle/FastDeploy.git
cd examples/vision/classification/yolov5cls/python/ cd examples/vision/classification/yolov5cls/python/
#下载yolov5cls模型文件和测试图片 #下载 YOLOv5Cls 模型文件和测试图片
wget https://bj.bcebos.com/paddlehub/fastdeploy/yolov5n-cls.onnx 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 wget https://gitee.com/paddlepaddle/PaddleClas/raw/release/2.4/deploy/images/ImageNet/ILSVRC2012_val_00000010.jpeg