apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2025-10-07 01:22:59 +08:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

[Model] Refactor PaddleClas module (#505)

Browse Source 
* Refactor the PaddleClas module

* fix bug

* remove debug code

* clean unused code

* support pybind

* Update fd_tensor.h

* Update fd_tensor.cc

* temporary revert python api

* fix ci error

* fix code style problem
This commit is contained in:
Jason
2022-11-07 19:33:47 +08:00
committed by GitHub
parent a0a8ace174
commit 3589c0fa94
15 changed files with 527 additions and 142 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -14,8 +14,9 @@
from __future__ import absolute_import
from .contrib.yolov5cls import YOLOv5Cls
from .ppcls import PaddleClasModel
from .ppcls import *
from .contrib.resnet import ResNet
PPLCNet = PaddleClasModel
PPLCNetv2 = PaddleClasModel
EfficientNet = PaddleClasModel