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

Validate all backends for detection models and add demo code & docs (#94)

Browse Source 
* Validate all backends for detection models and add demo code and doc

* Delete .README.md.swp
This commit is contained in:
Jason
2022-08-11 10:03:53 +08:00
committed by GitHub
parent d8f312b0a0
commit 3e01118d01
32 changed files with 1351 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
csrc/fastdeploy/pybind/fastdeploy_runtime.cc
View File

@@ -28,6 +28,8 @@ void BindRuntime(pybind11::module& m) {
.def("use_trt_backend", &RuntimeOption::UseTrtBackend)
.def("enable_paddle_mkldnn", &RuntimeOption::EnablePaddleMKLDNN)
.def("disable_paddle_mkldnn", &RuntimeOption::DisablePaddleMKLDNN)
.def("enable_paddle_log_info", &RuntimeOption::EnablePaddleLogInfo)
.def("disable_paddle_log_info", &RuntimeOption::DisablePaddleLogInfo)
.def("set_paddle_mkldnn_cache_size",
&RuntimeOption::SetPaddleMKLDNNCacheSize)
.def("set_trt_input_shape", &RuntimeOption::SetTrtInputShape)