一种基于注意力机制和关系网络的自动驾驶图像识别方法

摘要

本发明提出了一种基于注意力机制和关系网络的自动驾驶图像识别方法，首先，使用卷积神经网络提取自动驾驶图像特征；接着，通过注意力机制计算注意力特征；然后，利用关系网络生成图像中对象之间的关系结果；最后通过损失函数训练网络，并用训练好的模型进行自动驾驶场景的关系判断。本发明运用关系网络识别自动驾驶图像，能显式表达图像中对象之间的关系，从而提高识别和推理的准确性；此外，引入了新的注意力机制，降低了模型在处理复杂自动驾驶场景时的计算复杂度，提高了模型有效性和学习效率。