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

add yolov6 c++ and yolov6 pybind (#16)

Browse Source 
* update .gitignore

* Added checking for cmake include dir

* fixed missing trt_backend option bug when init from trt

* remove un-need data layout and add pre-check for dtype

* changed RGB2BRG to BGR2RGB in ppcls model

* add model_zoo yolov6 c++/python demo

* fixed CMakeLists.txt typos

* update yolov6 cpp/README.md
This commit is contained in:
DefTruth
2022-07-14 16:12:28 +08:00
committed by GitHub
parent 5f83b3c532
commit de7c06a309
27 changed files with 955 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
fastdeploy/vision/vision_pybind.cc
View File

@@ -18,6 +18,7 @@ namespace fastdeploy {
void BindPpClsModel(pybind11::module& m);
void BindUltralytics(pybind11::module& m);
void BindMeituan(pybind11::module& m);
#ifdef ENABLE_VISION_VISUALIZE
void BindVisualize(pybind11::module& m);
#endif
@@ -40,6 +41,9 @@ void BindVision(pybind11::module& m) {
BindPpClsModel(m);
BindUltralytics(m);
BindMeituan(m);
#ifdef ENABLE_VISION_VISUALIZE
BindVisualize(m);
#endif
}
} // namespace fastdeploy