English | 简体中文
- YOLOv5Cls v6.2 model deployment is based on YOLOv5 and Pre-trained Models on ImageNet.
- （1）The *-cls.pt model provided by Official Repository can export the ONNX file using
export.pyin YOLOv5, then deployment can be conducted; - （2）The YOLOv5Cls v6.2 Model trained by personal data should export the ONNX file using
export.pyin YOLOv5.
- （1）The *-cls.pt model provided by Official Repository can export the ONNX file using
For developers' testing, models exported by YOLOv5Cls are provided below. Developers can download them directly. (The model accuracy in the following table is derived from the source official repository)
| Model | Size | Accuracy(top1) | Accuracy(top5) |
|---|---|---|---|
| YOLOv5n-cls | 9.6MB | 64.6% | 85.4% |
| YOLOv5s-cls | 21MB | 71.5% | 90.2% |
| YOLOv5m-cls | 50MB | 75.9% | 92.9% |
| YOLOv5l-cls | 102MB | 78.0% | 94.0% |
| YOLOv5x-cls | 184MB | 79.0% | 94.4% |
- Document and code are based on YOLOv5 v6.2.