利索能及
我要发布
收藏
专利号: 2020103867343
申请人: 西安理工大学
专利类型:发明专利
专利状态:已下证
更新日期:2026-10-22
缴费截止日期: 暂无
联系人

摘要:

权利要求书:

1.一种基于旋转不变卷积神经网络的书法字风格分类方法,其特征在于,包括以下步骤:步骤1,建立取样样本并预处理数据集;

步骤2,划分数据集为训练样本集和测试样本集,并对训练集样本进行数据扩充作为新的训练集;

步骤3,对于步骤2得到的训练集中的每一张训练样本,提取16个副本图像,并将他们一一对应输入到完全相同的16个卷积架构中处理,将处理后得到的16个特征图串连接起来,输入到卷积架构后面的3个串联的全连接层中,得到该训练样本的分类结果;

步骤4,将步骤3得到的分类结果带入到交叉熵损失函数中,计算出本次训练产生的损失,然后求出损失函数关于网络中权值参数的梯度,引入Nesterov动量后使用梯度下降算法不断更新模型的权值参数,使损失函数达到全局最小值,从而得到训练好的网络模型;

步骤5,将待分类测试样本集输入到步骤4训练好的模型中进行书法字风格分类,得到分类准确率。

2.根据权利要求1所述的一种基于旋转不变卷积神经网络的书法字风格分类方法,其特征在于,步骤1的具体操作为:步骤1.1,用最小边框算法从整幅图像中获取单个书法字符;

步骤1.2,将得到的不同书法风格的每张图像按其中心对齐,并将它们裁剪成相同的大小64×64×3,得到取样样本。

3.根据权利要求2所述的一种基于旋转不变卷积神经网络的书法字风格分类方法,其特征在于,步骤2的具体操作为:步骤2.1,对步骤1获得的取样样本以3:1的比例随机分成训练样本集和测试样本集;

步骤2.2,对训练样本集进行数据扩充操作:即通过随机扰动图像来扩展训练样本集,每个训练样本集都以五种方式进行随机扰动,这五种数据扩充的方式为旋转、平移、缩放、翻转以及图像增亮。

4.根据权利要求3所述的一种基于旋转不变卷积神经网络的书法字风格分类方法,其特征在于,步骤2中,五种数据扩充的方式:旋转、平移、缩放、翻转以及图像增白的具体要求如下:旋转:随机旋转一个角度,该角度为在0°到360°之间均匀采样;

平移:在x和y方向上移动,移动的距离在‑4到4个像素之间均匀采样;移动的大小是有限的,以确保感兴趣的对象仍然在图像的中心;

缩放:使用缩放因子随机缩放,随机因子在0.9到1.1之间均匀采样;

翻转:图像以0.5的概率随机翻转;

增亮:使用PCA来改变RGB通道的强度,即给每个像素加上[P1,P2,P3][a1λ1,α2λ2,α3λ3];

Pi和λi是3×3RGB像素值协方差矩阵的第i个特征向量和特征值,αi为从均值为0,标准差为0.1的高斯分布中抽取的随机变量,i=1,2,3。

5.根据权利要求4所述的一种基于旋转不变卷积神经网络的书法字风格分类方法,其特征在于,步骤3的具体操作为:步骤3.1,提取副本图像的具体操作为:将训练样本不做任何变化作为第一子图像,将训练样本逆时针旋转45°作为第二子图像,将这两张子图像分别水平翻转又得到第三子图像及第四子图像,至此一共得到四张子图像;

对于每一张子图像,从子图像的左上角、右上角、左下角及右下角四个角落分别提取四个45×45×3大小的副本,对从子图像左上角提取到的副本不做变化,从子图像右上角提取到的副本逆时针旋转90°,从子图像左下角提取到的副本顺时针旋转90°,从子图像右下角提取到的副本顺时针旋转180°,使字体的中心在每个副本的右下角,至此从四张子图像中得到16个副本图像;

步骤3.2,将步骤3.1中得到的16个不同的副本分别输入到16个相同的卷积架构中进行处理得到16个不同的特征图;

步骤3.3,将步骤3.2中由卷积架构处理得到的16张特征图,串连接起来输入到后面的3个全连接层中,其中,第一个和第二个是maxout层,第三个是softmax线性层,由softmax线性层得到预测的分数向量S,再通过softmax线性层通过将每个类别得到的分数压缩到[0,

1]之间来表示分类结果的概率,如公式(1)所示:

其中j为训练样本,S为全连接层计算得到的分数向量,训练样本的正确标签对应的预测分数为Sj,N是标签种类的数量,pj是分类结果的概率。

6.根据权利要求5所述的一种基于旋转不变卷积神经网络的书法字风格分类方法,其特征在于,步骤4的具体操作为:步骤4.1,利用公式(1)得到的分类结果pj,通过公式(2)交叉熵损失函数,计算出本次训练产生的损失,如公式(2)所示,L=‑∑yjlog(pj) (2)

公式(2)中yj为训练样本的真实分布概率,pj为公式(1)中模型预测出正确结果的概率,交叉熵随着网络分类结果pj的增大而减小;

步骤4.2,求出交叉熵损失函数关于网络中权值参数的梯度,引入Nesterov动量计算出累计梯度,如公式(3)所示:公式(3)中V′为上一次训练的累计梯度,V为本次训练的累计梯度, 为本次交叉熵损失函数关于权值参数W的梯度,β为动量参数,β=0.9;

步骤4.3,根据公式(3)得到的本次累计梯度V,使用梯度下降算法不断更新权值参数,最终得到使损失函数最小的一组权值参数作为我们训练好的模型参数,网络参数的更新如公式(4)所示:W=W′‑αV (4)

式(4)中W为本次训练得到的权值参数,W′为上一次训练的权值参数,α为学习率,α=

0.0004,V为本次训练的累计梯度;

步骤4.4,不断迭代步骤4.3,得到使损失函数最小的一组权值参数作为训练好的模型参数。

7.根据权利要求6所述的一种基于旋转不变卷积神经网络的书法字风格分类方法,其特征在于,步骤5的具体操作为:将测试样本集分为16个组,每组50张图像,迭代计算每个组的分类准确率,最后求得16个组的平均准确率即为测试集样本分类的最终准确率。