Add YOLOv5-cls Model (#335)

* add yolov5cls

* fixed bugs

* fixed bugs

* fixed preprocess bug

* add yolov5cls readme

* deal with comments

* Add YOLOv5Cls Note

* add yolov5cls test

Co-authored-by: Jason <jiangjiajun@baidu.com>
This commit is contained in:
WJJ1995
2022-10-12 15:57:26 +08:00
committed by GitHub
parent 945e197bd1
commit b557dbc2d8
19 changed files with 719 additions and 4 deletions

View File

@@ -55,7 +55,7 @@ PaddleClas模型加载和初始化其中model_file, params_file为训练模
> PaddleClasModel.predict(input_image, topk=1)
> ```
>
> 模型预测结口,输入图像直接输出检测结果。
> 模型预测结口,输入图像直接输出分类topk结果。
>
> **参数**
>