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

Rename fastdeploy_runtime.h to runtime.h and Frontend to ModelFormat (#263)

Browse Source 
rename frontend to model_format
This commit is contained in:
Jason
2022-09-22 13:24:05 +08:00
committed by GitHub
parent 9bebdaa915
commit e227c5625e
126 changed files with 400 additions and 380 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
fastdeploy/vision/detection/contrib/yolov5lite_pybind.cc
View File

@@ -18,7 +18,8 @@ namespace fastdeploy {
void BindYOLOv5Lite(pybind11::module& m) {
pybind11::class_<vision::detection::YOLOv5Lite, FastDeployModel>(m,
"YOLOv5Lite")
.def(pybind11::init<std::string, std::string, RuntimeOption, Frontend>())
.def(pybind11::init<std::string, std::string, RuntimeOption,
ModelFormat>())
.def("predict",
[](vision::detection::YOLOv5Lite& self, pybind11::array& data,
float conf_threshold, float nms_iou_threshold) {