Electronics Optics & Control, Volume. 31, Issue 9, 70(2024)
An Improved HighPerformance Object Detector Based on YOLOv7-tiny
Aiming at the problems of large amount of network parameters and low detection accuracy of the existing YOLOseries object detectorsa highperformance universal object detector named YOLOv7TT is proposed based on YOLOv7tiny model.FirstlyGeneralized and Friendly ELAN (GFELAN) module is introduced into Backbone and Neck networks to expand the width and depth of the network and eliminate the redundant features generated by the networkso as to reduce the parameter quantity and computation cost.Thenthe improved SimOTA sample allocation method is used to optimize the allocation of positive samples in the training process and accelerate the convergence speed of the network.Finallythe knowledge distillation method is used to distill and train the model to improve its detection accuracy while ensuring lightweight.The experimental results show that:1) Compared with YOLOv7tinyYOLOv7TT reduces the quantity of network parameters by 11% and 9.7%and improves the AP by 4.2 and 3.0 percentage points respectively on the VOC2007 and COCO2017 datasets;and 2) The model detection accuracy is further improved by using knowledge distillationthe AP reaches 59.4% (with 5.3 percentage points improved) on the VOC2007 datasetwhich effectively solves the problems of large quantity parameters and low detection accuracy.
Get Citation
Copy Citation Text
ZHU Wenxu, SHI Tao, ZHOU Jiarun, LIU Zulin, LIU Haixin. An Improved HighPerformance Object Detector Based on YOLOv7-tiny[J]. Electronics Optics & Control, 2024, 31(9): 70
Received: Oct. 16, 2023
Accepted: --
Published Online: Oct. 22, 2024
The Author Email: