English | 简体中文
The YOLOv7End2EndORT deployment is based on YOLOv7branch code and Pre-trained Model Based on COCO. Attention: YOLOv7End2EndORT is designed for the inference of exported End2End models in the ORT_NMS version in YOLOv7. YOLOv7 class is for the inference of models without nms. YOLOv7End2EndTRT is for the inference of End2End models in the TRT_NMS version.
- （1）*.pt provided by Official Repository should Export the ONNX Model to complete the employment. The deployment of *.trt and *.pose models is not supported.
- （2）The YOLOv7 model trained by personal data should Export the ONNX Model. Refer to Detailed Deployment Documents to complete the deployment.
# Download yolov7 model files
wget https://github.com/WongKinYiu/yolov7/releases/download/v0.1/yolov7.pt
# Export files in onnx format with ORT_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 --max-wh 640
# The commands for exporting other models are similar Replace yolov7.pt with yolov7x.pt yolov7-d6.pt yolov7-w6.pt ...For developers' testing, models exported by YOLOv7End2EndORT 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-ort-nms | 141MB | 51.4% | This model file is sourced from YOLOv7，GPL-3.0 License |
| yolov7x-end2end-ort-nms | 273MB | 53.1% | This model file is sourced from YOLOv7，GPL-3.0 License |
| yolov7-w6-end2end-ort-nms | 269MB | 54.9% | This model file is sourced from YOLOv7，GPL-3.0 License |
| yolov7-e6-end2end-ort-nms | 372MB | 56.0% | This model file is sourced from YOLOv7，GPL-3.0 License |
| yolov7-d6-end2end-ort-nms | 511MB | 56.6% | This model file is sourced from YOLOv7，GPL-3.0 License |
| yolov7-e6e-end2end-ort-nms | 579MB | 56.8% | This model file is sourced from YOLOv7，GPL-3.0 License |
- Document and code are based on YOLOv7 0.1