apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2025-10-30 11:26:39 +08:00
Code Issues Actions 6 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

1
fastdeploy/vision/detection/__init__.py
View File

@@ -21,5 +21,6 @@ from .contrib.yolox import YOLOX
from .contrib.yolov5 import YOLOv5
from .contrib.yolov5lite import YOLOv5Lite
from .contrib.yolov6 import YOLOv6
from .contrib.yolov7end2end_trt import YOLOv7End2EndTRT
from .contrib.yolov7end2end_ort import YOLOv7End2EndORT
from .ppdet import PPYOLOE, PPYOLO, PPYOLOv2, PaddleYOLOX, PicoDet, FasterRCNN, YOLOv3