mirror of
https://github.com/PaddlePaddle/FastDeploy.git
synced 2025-10-13 04:13:58 +08:00
02bd22422e
* add GPL lisence * add GPL-3.0 lisence * add GPL-3.0 lisence * add GPL-3.0 lisence * support yolov8 * add pybind for yolov8 * add yolov8 readme Co-authored-by: DefTruth <31974251+DefTruth@users.noreply.github.com>
English | 简体中文
YOLOv7End2EndTRT Ready-to-deploy Model
The YOLOv7End2EndTRT deployment is based on YOLOv7 branch code and Pre-trained Model Baesd on COCO. Attention: YOLOv7End2EndTRT is designed for the inference of exported End2End models in the TRT_NMS version in YOLOv7. YOLOv7 class is for the inference of models without nms. YOLOv7End2EndORT is for the inference of End2End models in the ORT_NMS version.
- （1）*.pt provided by Official Repository should Export the ONNX Model to complete the deployment. The deployment of *.trt and *.pose models is not supported.
- （2）The YOLOv7 model trained by personal data should Export the ONNX Model. Please refer to Detailed Deployment Documents to complete the deployment.
Export the ONNX Model
# Download yolov7 Model Files
wget https://github.com/WongKinYiu/yolov7/releases/download/v0.1/yolov7.pt
# Export file in onnx format with TRT_NMS (Tips: corresponding to the code of YOLOv7 release v0.1)
python export.py --weights yolov7.pt --grid --end2end --simplify --topk-all 100 --iou-thres 0.65 --conf-thres 0.35 --img-size 640 640
# The commands for exporting other models are similar. Replace yolov7.pt with yolov7x.pt yolov7-d6.pt yolov7-w6.pt ...
# Only onnx files are required to employ YOLOv7End2EndTRT. Additional trt files are not required because automatic switching happens during inference.
Download Pre-trained ONNX Models
For developers' testing, models exported by YOLOv7End2EndTRT are provided below. Developers can download them directly. (The accuracy in the following table is derived from the source official repository)
| Model | Size | Accuracy | Note |
|---|---|---|---|
| yolov7-end2end-trt-nms | 141MB | 51.4% | This model file is sourced from YOLOv7，GPL-3.0 License |
| yolov7x-end2end-trt-nms | 273MB | 53.1% | This model file is sourced from YOLOv7，GPL-3.0 License |
| yolov7-w6-end2end-trt-nms | 269MB | 54.9% | This model file is sourced from YOLOv7，GPL-3.0 License |
| yolov7-e6-end2end-trt-nms | 372MB | 56.0% | This model file is sourced from YOLOv7，GPL-3.0 License |
| yolov7-d6-end2end-trt-nms | 511MB | 56.6% | This model file is sourced from YOLOv7，GPL-3.0 License |
| yolov7-e6e-end2end-trt-nms | 579MB | 56.8% | This model file is sourced from YOLOv7，GPL-3.0 License |
Detailed Deployment Documents
Release Note
- Document and code are based on YOLOv7 0.1