Abstract
In response to the requirements for detection accuracy and speed in security detection technology in current smart city construction, a model compression method based on YOLOv5s is proposed, aiming to lighten the original model and detect the opening and closing states of car windows. Firstly, a structured pruning method based on channel weights is employed to perform channel pruning. Then, a fine-tuning training method based on logical distillation is adopted to restore the network detection accuracy. Experimental results on the car window dataset show that compared with the original algorithm, the improved algorithm only loses 0.3% in detection accuracy while reducing the number of parameters and computations to 41.5% and 44.3% of the original model, respectively. This significantly reduces the model complexity and achieves model lightweighting.
Talk to us
Join us for a 30 min session where you can share your feedback and ask us any queries you have
Disclaimer: All third-party content on this website/platform is and will remain the property of their respective owners and is provided on "as is" basis without any warranties, express or implied. Use of third-party content does not indicate any affiliation, sponsorship with or endorsement by them. Any references to third-party content is to identify the corresponding services and shall be considered fair use under The CopyrightLaw.