[发明专利]图像分类模型训练方法及系统、图像分类方法及系统

摘要

本发明公开了一种图像分类模型训练方法及系统、图像分类方法及系统。该图像分类模型训练方法包括：获取预先训练好的图像分类模型以及训练数据集；在图像分类模型的卷积层和全连接层之间加入注意力机制模块，并根据分类标签的类别数量调整全连接层中输出节点的数量，以更新图像分类模型；使用训练数据集训练图像分类模型，以获得目标图像分类模型；其中，在训练过程中冻结全部或部分卷积层的权重参数。本发明将训练好的图像模型在训练数据集上进行迁移学习，使得该模型能够在过滤无关图片的同时准确识别图片类别，获得了相较于直接对整个模型进行训练的方式来说更优的性能。