In order to solve the problem that it is difficult to control the abuse of high beam at night, a high beam detection algorithm YOLO-CAU based on YOLOv4 is designed and implemented. Then, aiming at the problem that it is difficult deployed on devices with limited computing power with a complex neural network, we apply the model compression algorithm of knowledge distillation to the image object detection task, and conducts distillation training on the YOLOCAU model with the high beam data set, transferring the knowledge learned by YOLO-CAU to a simple network (YOLO-CAU-S) with less computation and faster detection speed. Compared with the existing knowledge distillation algorithm, the algorithm designed in this paper not only restricts the output of the student network to be close to the output of the teacher, but also proposes a multi-scale pixel attention module to construct the attention feature maps of the teacher model and student model, which further transfers the strong attention of the teacher network to the student. The experimental results show that the knowledge distillation algorithm designed in this paper can greatly improve the detection efficiency of the model on the premise of ensuring the detection accuracy, and the compressed model parameters are only 6% of the teacher network.
Access to the requested content is limited to institutions that have purchased or subscribe to SPIE eBooks.
You are receiving this notice because your organization may not have SPIE eBooks access.*
*Shibboleth/Open Athens users─please
sign in
to access your institution's subscriptions.
To obtain this item, you may purchase the complete book in print or electronic format on
SPIE.org.
INSTITUTIONAL Select your institution to access the SPIE Digital Library.
PERSONAL Sign in with your SPIE account to access your personal subscriptions or to use specific features such as save to my library, sign up for alerts, save searches, etc.