World Electric Vehicle Journal • Vol 15 • No 8
Vehicle Classification Algorithm Based on Improved Vision Transformer
July 2024 • Xinlong Dong, Peicheng Shi, Yueyue Tang, Li Yang, Aixi Yang, Taonian Liang
Vehicle classification technology is one of the foundations in the field of automatic driving. With the development of deep learning technology, visual transformer structures based on attention mechanisms can represent global information quickly and effectively. However, due to direct image segmentation, local feature details and information will be lost. To solve this problem, we propose an improved vision transformer vehicle classification network (IND-ViT). Specifically, we first design a CNN-In D branch module…