How to Prune and Sparse YOLOv5 . Yolov5 is a modern object detection…

Yolov5 is a modern object detection algorithm, that has been written in a PyTorch, Besides this, it's having, fast speed, high accuracy, easy to install and use. The importance of yolov5 raised, due to its different export and deployments modules. We can convert trained model (.pt) in many extensions i.e., Pruning is the process of "Modification of weights by reducing the weights parameters, that does not impact on classification of classes". Yolov5 is using technique of pruning in a way that "randomly in some percentage of weights parameters, nn.conv2d layers whose weights are nearly zero, it will converted them to zeros". You will first need to train yolov5 on your custom data, once training will be completed, then you need to validate your trained model on Validation data, at that time you can call the prune function for the trained model to prune.