[发明专利]图像识别模型的训练方法、装置、计算机设备及存储介质

摘要

本发明公开了一种图像识别模型的训练方法，应用于人工智能技术领域，用于解决目前的训练方法在对图像识别模型进行训练时对GPU资源的占用率高的技术问题。本发明提供的方法包括：通过图像识别模型的多通道卷积层将样本图像转换为多通道特征图；按照预设的通道比例将多通道特征图拆分为初始特征图和第一特征图；将第一特征图输入至图像识别模型的主干网络进行前向传播得到中间特征图；将中间特征图与初始特征图进行拼接得到训练样本特征图；将训练样本特征图输入至图像识别模型的全连接层，得到训练样本特征图的映射类别；以样本图像所属的实际类别作为目标调整主干网络的参数，通过样本图像对图像识别模型进行循环训练，得到训练好的图像识别模型。