FastDeploy/examples/vision/detection/yolov5lite

English | 简体中文

YOLOv5Lite Ready-to-deploy Model

• The YOLOv5Lite Deployment is based on the code of YOLOv5-Lite and Pre-trained Model Based on COCO。

  • （1）The *.pt provided by Official Repository should Export the ONNX Modelto complete the deployment；
  • （2）The YOLOv5Lite model trained by personal data should Export the ONNX Model. Refer to Detailed Deployment Documents to complete the deployment.

Export the ONNX Model

• Auto-acquisition Visit official YOLOv5Lite github repository, follow the guidelines to download the yolov5-lite-xx.onnx model(Tips: The official ONNX files are currently provided without the decode module)

  # Download yolov5-lite model files(.onnx)
  Download from https://drive.google.com/file/d/1bJByk9eoS6pv8Z3N4bcLRCV3i7uk24aU/view
  Official Repo also supports Baidu cloud download
  

• Manual Acquisition

  Visit official YOLOv5Lite github repository, follow the guidelines to download the yolov5-lite-xx.pt model, and employ export.py to get files in onnx format.

  • Export ONNX files with the decode module

  First refer to YOLOv5-Lite#189 to modify the code.

  # Download yolov5-lite model files(.pt)
  Download from https://drive.google.com/file/d/1oftzqOREGqDCerf7DtD5BZp9YWELlkMe/view
  Official Repo also supports Baidu cloud download
  
  # Export files in onnx format
  python export.py --grid --dynamic --concat --weights PATH/TO/yolov5-lite-xx.pt
  
  
  

  • Export ONNX files without the docode module(No code changes are required)

  # Download yolov5-lite model files
  Download from https://drive.google.com/file/d/1oftzqOREGqDCerf7DtD5BZp9YWELlkMe/view
  Official Repo also supports Baidu cloud download
  
  # Export files in onnx format
  python export.py --grid --dynamic --weights PATH/TO/yolov5-lite-xx.pt
  
  

Download Pre-trained ONNX Model

For developers' testing, models exported by YOLOv5Lite 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
YOLOv5Lite-e	3.1MB	35.1%	This model file is sourced from YOLOv5-Lite，GPL-3.0 License
YOLOv5Lite-s	6.3MB	42.0%	This model file is sourced from YOLOv5-Lite，GPL-3.0 License
YOLOv5Lite-c	18MB	50.9%	This model file is sourced fromYOLOv5-Lite，GPL-3.0 License
YOLOv5Lite-g	21MB	57.6%	This model file is sourced from YOLOv5-Lite，GPL-3.0 License

Detailed Deployment Documents

• Python Deployment
• C++ Deployment

Release Note

• Document and code are based on YOLOv5-Lite v1.4