To address the issue that temporal features and spatial factors of the traffic environment affect the accuracy of vehicle trajectory prediction in vehicle driving, this paper proposes a vehicle trajectory prediction method integrating temporal multi-head self-attention and social pooling based on the Social Generative Adversarial Network (SMA-GAN). Firstly, the historical trajectory features are extracted by the temporal correlation of the target vehicle’s own trajectory data using the multi-head self-attention mechanism. Then, the spatial dimensional features of the target vehicle are extracted by the social pooling mechanism based on the spatial positional relationship between the target vehicle and the surrounding vehicles. Finally, the predicted trajectory of the target vehicle is obtained by the encoder-decoder. Model training and comparison tests are conducted using the NGSIM dataset, and the results show that the SMA-GAN model has higher prediction accuracy and efficiency in the highway scene.
| 科 Family | 属数 Number of genus | 种数 Number of species | 占总种数比例 Percentage of total species (%) | 属 Genus | 种数 Number of species | 占总种数比例 Percentage of total species (%) |
|---|---|---|---|---|---|---|
| 鹅膏菌科Amanitaceae | 2 | 11 | 5.26 | 鹅膏菌属 Amanita | 10 | 4.78 |
| 小菇科 Mycenaceae | 2 | 12 | 5.74 | 丝盖伞属 Inocybe | 5 | 2.39 |
| 多孔菌科 Polyporaceae | 8 | 14 | 6.70 | 蜡蘑属 Laccaria | 5 | 2.39 |
| 红菇科 Russulaceae | 3 | 23 | 11.00 | 小皮伞属 Marasmius | 6 | 2.87 |
| 小菇属 Mycena | 11 | 5.26 | ||||
| 光柄菇属 Pluteus | 5 | 2.39 | ||||
| 红菇属 Russula | 17 | 8.13 | ||||
| 栓菌属 Trametes | 5 | 2.39 |