【技术实现步骤摘要】
识别分类模型训练方法和系统
本申请涉及深度学习
,特别是涉及一种识别分类模型训练方法和识别分类模型训练系统。
技术介绍
人工智能发展迅速,其中,卷积神经网络模型是深度学习中重要的模型,主要应用在图片处理,其在图片处理以及分类上有着显著的效果。传统技术下,用于分类识别的卷积神经网络模型训练过程中进行分类的图片一般是是无损且特征明显的图片,训练后的卷积神经网络模型适用于处理无损且特征明显的图片。但是在实际应用环境中,由于信号受到干扰,或者设备存在质量问题等,所获取的图片往往会存在噪声,或者出现特征不明显的现象,用于分类识别的卷积神经网络模型对具有噪声或者特征不明显的图片进行识别的准确度低。
技术实现思路
基于此,有必要针对上述用于分类识别的卷积神经网络模型对具有噪声或者特征不明显的图片进行识别的准确度低的问题,提供一种识别分类模型训练方法和系统。一种识别分类模型训练方法,包括以下步骤:根据破坏率和初始图片样本数据,获取有损图片样本数据;将有损图片样本数据输入去噪声模型中进行训练,获取还原图片样本数据和去噪损失值;将还原图片样本数据输入识别分类模型中进行训练,获取样本标签值和识别损失值;根据样本标签值、去噪损失值和识别损失值,分别对去噪声模型和识别分类模型进行反向传导算法的计算,获取训练好的去噪声模型和训练好的识别分类模型。一种识别分类模型训练系统,包括:有损图片样本数据获取模块,用于根据破坏率和初始图片样本数据,获取有损图片样本数据;还原图片样本数据获取模块,用于将有损图片样本数据输入去噪声模型中进行训练,获取还原图片样本数据和去噪损失值;样本标签值获取模块 ...
【技术保护点】
1.一种识别分类模型训练方法,其特征在于,包括以下步骤:根据破坏率和初始图片样本数据,获取有损图片样本数据;将所述有损图片样本数据输入去噪声模型中进行训练,获取还原图片样本数据和去噪损失值;将所述还原图片样本数据输入识别分类模型中进行训练,获取样本标签值和识别损失值;根据所述样本标签值、所述去噪损失值和所述识别损失值,分别对所述去噪声模型和所述识别分类模型进行反向传导算法的计算,获取训练好的去噪声模型和训练好的识别分类模型。
【技术特征摘要】
1.一种识别分类模型训练方法,其特征在于,包括以下步骤:根据破坏率和初始图片样本数据,获取有损图片样本数据;将所述有损图片样本数据输入去噪声模型中进行训练,获取还原图片样本数据和去噪损失值;将所述还原图片样本数据输入识别分类模型中进行训练,获取样本标签值和识别损失值;根据所述样本标签值、所述去噪损失值和所述识别损失值,分别对所述去噪声模型和所述识别分类模型进行反向传导算法的计算,获取训练好的去噪声模型和训练好的识别分类模型。2.根据权利要求1所述的识别分类模型训练方法,其特征在于,所述根据破坏率和初始图片样本数据,获取有损图片样本数据的步骤,包括以下步骤:根据所述破坏率获取随机值和预设范围;在所述初始图片样本数据中所述预设范围的数据值上叠加所述随机值,并生成所述有损图片样本数据。3.根据权利要求1所述的识别分类模型训练方法,其特征在于,所述去噪声模型包括卷积层、下采样层、还原层、融合还原层和全连接层;所述将所述有损图片样本数据输入去噪声模型中进行训练,获取还原图片样本数据和去噪损失值的步骤,包括以下步骤:将所述有损图片样本数据输入所述卷积层,并提取所述有损图片样本数据的特征;将所述有损图片样本数据的特征输入所述下采样层,并获取深化特征;将所述深化特征输入所述还原层,并获取还原数据;将所述还原数据输入所述融合还原层并获取融合还原数据,将所述融合还原数据输入所述全连接层,并获取所述还原图片样本数据;根据所述初始图片样本数据和所述还原图片样本数据,获取所述去噪损失值。4.根据权利要求3所述的识别分类模型训练方法,其特征在于,所述破坏率包括第一破坏率和第二破坏率,所述有损图片样本数据包括第一有损图片样本数据和第二有损图片样本数据,所述根据破坏率和初始图片样本数据,获取有损图片样本数据的步骤,包括以下步骤:根据所述第一破坏率和所述初始图片样本数据生成所述第一有损图片样本数据,根据所述第二破坏率和所述初始图片样本数据生成所述第二有损图片样本数据;所述卷积层包括第一卷积层、第二卷积层和第三卷积层,所述有损图片样本数据的特征包括所述第一有损图片样本数据的第一特征、所述第二有损图片样本数据的第二特征和第三特征,所述将所述有损图片样本数据输入所述卷积层,并提取所述有损图片样本数据的特征的步骤,包括以下步骤:将所述第一有损图片样本数据和所述第一破坏率输入所述第一卷积层,并提取所述第一有损图片样本数据的第一特征;将所述第二有损图片样本数据和所述第二破坏率输入所述第二卷积层,并提取所述第二有损图片样本数据的第二特征;将所述第一特征和所述第二特征输入所述第三卷积层,并提取所述第三特征;所述下采样层包括第一下采样层、第二下采样层和第三下采样层,所述深化特征包括第一深化特征、第二深化特征和第三深化特征,所述将所述有损图片样本数据的特征输入所述下采样层,并获取深化特征的步骤,包括以下步骤:将所述第一特征输入所述第一下采样层并获取所述第一深化特征,将所述第二特征输入所述第二下采样层并获取所述第二深化特征,将所述第三特征输入所述第三下采样层并获取所述第三深化特征;所述还原层包括第一还原层、第二还原层和第三还原层,所述还原数据包括第一还原数据、第二还原数据和第三还原数据,所述将所述深化特征输入所述还原层,并获取还原数据的步骤,包括以下步骤:将所述第一深化特征输入所述第一还原层并获取所述第一还原数据,将所述第二深化特征输入所述第二还原层并获取所述第二还原数据,将所述第三深化特征输入所述第三还原层并获取所述第三还原数据;所述融合还原层包括第一融合还原层和第二融合还原层,所述融合还原数据包括第一融合还原数据和第二融合还原数据,所述将所述还原数据输入所述融合还原层并获取融合还原数据,将所述融合还原数据输入所述全连接层,并获取所述还原图片样本数据的步骤,包括以下步骤:将所述第一还原数据和所述第三还原数据输入所述第一融合还原层,并获取所述第一融合还原数据;将所述第二还原数据和所述第三还原数据输入所述第二融合还原层,并获取所述第二融合还原数据;将所述第一融合还原数据和第二融合还原数据输入所述全连接层,并获取所述还原图片样本数据。5.根据权利要求3所述的识别分类模型训练方法,其特征在于,所述根据所述初始图片样本数据和所述还原图片样本数据,获取所述去噪损失值的步骤,包括以下步骤:根据获取所述去噪损失值,其中,q(x,z)为所述去噪损失值,x为所述初始图片样本数据,z为所述还原图片样本数据,xi为所述初始图片样本数据中第i个数据值,zi为所述第i个数据值对应的还原图片样本中的数据值,d为所述初始图片样本数据的数据值的总个数。6.根据权利要求1所述的识别分类模型训练方法,其特征在于,所述识别分类模型包括一级卷积层、一级下采集层、二级卷积层、二级下采集层和一级全连接层;所述将所述还原图片样本数据输入识别分类模型中进行训练,获取样本标签值和识别损失值的步骤,包括以下步骤:将所述还原图片样本数据输入所述一级卷积层,并获取所述还原图片样本数据的一级提取特征;将所述一级提取特征输入所述一级下采集层,并获取二级提取特征;将所述二级提取特征输入所述二级卷积层,并获取三级提取特征;将所述三级提取特征输入所述二...
【专利技术属性】
技术研发人员:徐智浩,王达,周雪峰,苏泽荣,鄢武,
申请(专利权)人:广东省智能制造研究所,
类型:发明
国别省市:广东,44
还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。