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

[feature][vision] Add YOLOv7 End2End model with TRT NMS (#157)

Browse Source 
* [feature][vision] Add YOLOv7 End2End model with TRT NMS

* [docs] update yolov7end2end_trt examples docs

Co-authored-by: Jason <jiangjiajun@baidu.com>
This commit is contained in:
DefTruth
2022-08-30 15:02:48 +08:00
committed by GitHub
parent 30bb233db8
commit 3c1330e896
14 changed files with 902 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
csrc/fastdeploy/vision/detection/detection_pybind.cc
View File

@@ -25,6 +25,7 @@ void BindYOLOv5(pybind11::module& m);
void BindYOLOX(pybind11::module& m);
void BindNanoDetPlus(pybind11::module& m);
void BindPPDet(pybind11::module& m);
void BindYOLOv7End2EndTRT(pybind11::module& m);
void BindYOLOv7End2EndORT(pybind11::module& m);
void BindDetection(pybind11::module& m) {
@@ -39,6 +40,7 @@ void BindDetection(pybind11::module& m) {
BindYOLOv5(detection_module);
BindYOLOX(detection_module);
BindNanoDetPlus(detection_module);
BindYOLOv7End2EndTRT(detection_module);
BindYOLOv7End2EndORT(detection_module);
}
} // namespace fastdeploy