关键词: YOLOv7-tiny, Ghost, 多尺度融合模块, 迁移学习, 边缘设备部署

Abstract: To address the common issues of excessive parameters,slow inference speed,limited detection performance,and difficulty in deploying neural networks on edge devices,this paper proposes an improved YOLOv7-tiny algorithm.Firstly,according to the characteristics of the original algorithm model structure,Ghost-ELAN module is introduced to compress the model greatly.Secondly,Ghost Bottleneck-2 is used to replace the convolution of the Neck part of the network,which further reduces the scale of the model.Then,the multi-scale fusion module Ghost-SPPCSPC is used to improve the understanding of feature information of the model,and the output layer convolution is replaced by GhostConv,which reduces the redundancy of common convolution and makes the maximum use of semantic information in the network.Finally,transfer learning is employed for enhancing generalized feature learning and improving performance of the model.Experimental results demonstrate that the improved model reduces parameter count and model size by 57.19% and 55.28%,respectively,achieving substantial compression over the original model while enhancing accuracy.With an inference speed of 278,the proposed model attains rapid,efficient,and lightweight objectives,making it highly suitable for deployment on edge devices.

Key words: YOLOv7-tiny, Ghost, multi-scale fusion module, transfer learning, edge device deployment