毫米波图像识别模型的训练方法、图像识别方法及装置

本公开提供一种毫米波图像识别模型的训练方法、图像识别方法及装置，方法包括：将上下文提取模块加入至识别模型中的主干网络中，以获取初始识别模型；获取训练数据；针对任一类型工具的毫米波图像，确定毫米波图像对应的目标图像特征图；基于目标图像特征图，确定毫米波图像对应的注意力输出，得到目标对象携带不同类型工具的毫米波图像对应的多个注意力输出，将多个注意力输出加入至初始识别模型，确定训练后的识别模型。本公开实施例提供的技术方案，通过对不同类型工具对应的毫米波图像进行训练，计算得到其对应的注意力输出，并得到不同类型工具对应的注意力输出，进而将注意力输出加入至初始识别模块中，大大提高了识别模型的训练效率。