apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2025-10-05 16:48:03 +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

2
fastdeploy/vision/classification/classification_pybind.cc
View File

@@ -16,11 +16,13 @@
namespace fastdeploy {
void BindYOLOv5Cls(pybind11::module& m);
void BindPaddleClas(pybind11::module& m);
void BindClassification(pybind11::module& m) {
auto classification_module =
m.def_submodule("classification", "Image classification models.");
BindYOLOv5Cls(classification_module);
BindPaddleClas(classification_module);
}
} // namespace fastdeploy