YOLOv5: Sparsifying to Improve Object Detection Performance

Neural Magic creates models and recipes that allow anyone to plug in their data and leverage SparseML’s recipe-driven approach on top of Ultralytics’ robust training pipelines for the popular YOLOv5 object detection networks. Sparsifying involves removing redundant information from neural networks using algorithms such as pruning and quantization, among others. This sparsification process results in faster inference and smaller file sizes for deployments.

To dive deeper into how we used sparsification and proprietary advancements in the DeepSparse Engine to enable GPU-level performance for YOLOv5 click through to our blog post – YOLOv5 on CPUs: Sparsifying to Achieve GPU-Level Performance and a Smaller Footprint.

This page walks through the following use cases for trying out the sparsified YOLOv5 models:

  • Compare the differences between the models for both accuracy and inference performance

  • Run the models for inference in deployment or applications

  • Train the models on new datasets

Video not loading? View full video here.

Sparse Models

YOLOv5l card not loading? View card here.
YOLOv5s card not loading? View card here.