基于迁移学习的图像分类在诗词中的应用研究

摘要

中国传统诗词中蕴含着丰富的文化内涵.为了从海量的诗词库中搜索出最符合图像意境的诗词,实现解析图像内容、提取图像特征关键词,结合项目需求,提出一种基于迁移学习的多EfficientNet融合网络的图像分类算法.收集、整理了基础诗词库,创建了项目专有的诗词意象图像数据集NID(nature image dataset),其中共有9大类;将在ImageNet图像数据集上训练好的EfficientNet模型迁移到NID中,对NID进行特征提取和图像标签匹配度的权值计算,结合每种图像类别训练得到的不同模型权重,融合9种模型权重,部署为一个多EfficientNet融合网络模型;最后对比了多种深度学习模型在NID上的表现性能.实验结果表明:多EfficientNet融合网络模型能够较为准确地解析图像,得到具有区分性的分类特征,并对NID的分类效果明显,收敛速度更快,精确率更高,符合项目中对诗词搜索的要求.