apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2025-12-24 13:28:13 +08:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Add YOLOv5-cls Model (#335)

Browse Source 
* 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
Download Patch File Download Diff File Expand all files Collapse all files

1
python/fastdeploy/vision/classification/__init__.py
View File

@@ -13,6 +13,7 @@
# limitations under the License.
from __future__ import absolute_import
from .contrib.yolov5cls import YOLOv5Cls
from .ppcls import PaddleClasModel
PPLCNet = PaddleClasModel