[发明专利]图像风格识别方法、装置、计算机设备及存储介质

摘要

本发明提供一种图像风格识别方法、装置、计算机设备及存储介质，图像识别方法包括：搭建卷积神经网络，将待识别图片输入卷积神经网络获取m幅卷积特征图，其中m大于1；将每幅卷积特征图压缩成n维向量以获取m个n维向量，对每两个n维向量进行内积计算获取m×m格拉姆矩阵；将m×m格拉姆矩阵输入到浅层网络中进行分类，输出待识别图片的图像风格。本发明技术方案通过提取中间特征图作为风格特征，获取了比全连接层更好的风格特征，并通过计算特征图的相关性特征，并且使用了一个浅层网络学习这个特征，获取了更好的风格表征，提高了分类性能，此过程实现了端对端学习，比人工抽取特征更加简便。