[发明专利]基于深度学习的图像识别方法、装置、设备及存储介质

摘要

本发明公开了一种基于深度学习的图像识别方法、装置、设备及存储介质。该方法包括：获取训练图像的类别标签和二值分割标签，对训练图像进行特征提取，得到特征空间；通过预设卷积神经网络，对特征空间进行注意力预测，得到目标热图，根据目标热图和二值分割标签，得到分割损失值；获取特征空间对应的嵌入向量，类别标签对应的类别中心，通过预设损失函数，计算类别中心与嵌入向量之间的距离损失值；对嵌入向量进行整合，得到图像类别，计算图像类别与类别标签的类别损失值，根据分割损失值，距离损失值和类别损失值，对预设卷积神经网络进行参数调整，以完成模型训练。本发明在使用深度学习进行图像识别，实现了训练时间更快和预测准确度更高。