【技术实现步骤摘要】
一种小批量图像识别方法、设备及介质
[0001]本专利技术涉及图像识别
,具体涉及一种小批量图像识别方法、设备及介质。
技术介绍
[0002]某些特殊领域(如战场、罕见病例医疗图像等)的图像由于难以获得、不易标注、重复性低等特点,只能获得小批量图像,导致可供利用的图像特征数据很少,从而增加了图像识别的难度。
[0003]现有技术通常采用深度学习对于图像进行识别,但是采用深度学习对于图像进行识别通常依赖于大量高质量的人工标注数据,在小批量图像识别任务中,由于显而易见的数据标注的困难性及数据本身的稀缺性,大量有标注图像数据难以获取。
[0004]其次,特殊领域的新技术装备不断更新,新的特殊领域目标也会不断涌现。而,由于新的特殊领域目标图像所采集到的数据存在不完整性和不确定性,导致可供利用的图像特征数据很少,采用深度学习难以对其进行识别。
[0005]因此,如何对小批量图像进行高精度识别成为研究热点。
技术实现思路
[0006]本专利技术所要解决的技术问题是由于小批量图像所采集到的数据存在不完整性和不确定性,导致难以对小批量图像进行精准识别,目的在于提供一种小批量图像识别方法、设备及介质,通过将采集到的为数不多的特殊领域目标图像作为原始图像输入,经过深度卷积神经网络提取特征,送入策略网络模型和价值网络模型加以推理决策,以实现端到端的小批量图像识别学习,解决了由于小批量图像所采集到的数据存在不完整性和不确定性,导致难以对小批量图像进行精准识别的问题。
[0007]本专利技 ...
【技术保护点】
【技术特征摘要】
1.一种小批量图像识别方法,其特征在于,包括如下步骤:S1、获取小批量图像;S2、构建特征提取网络模型,采用所述特征提取网络模型对所述小批量图像进行特征提取,得到所述小批量图像的图像特征;S3、构建小批量图像分类模型,将所述图像特征送入所述小批量图像分类模型中进行预测,得到分类标签预测;S4、根据分类标签预测对所述特征提取网络模型和所述小批量图像分类模型进行训练,得到小批量图像识别模型SDRL,采用所述小批量图像识别模型SDRL对所述小批量图像进行识别,得到目标图像。2.根据权利要求1所述的一种小批量图像识别方法,其特征在于,构建特征提取网络模型,采用所述特征提取网络模型对所述小批量图像进行特征提取,得到所述小批量图像的图像特征包括如下步骤:构建11层卷积神经网络M_Net;其中,所述卷积神经网络的前10层由卷积层和最大池化层交替构成,所述卷积神经网络M_Net的第11层由正则化层dropout构成;采用所述卷积层提取所述小批量图像的图像信息并通过ReLU函数进行激活后传递给所述最大池化层;采用所述最大池化层对所述图像信息进行挑选;采用所述正则化层dropout随机删除所述卷积神经网络M_Net中的部分神经元。3.根据权利要求1所述的一种小批量图像识别方法,其特征在于,构建小批量图像分类模型,将所述图像特征送入所述小批量图像分类模型中进行预测,得到分类标签预测包括如下步骤:构建策略网络模型,使用所述策略网络模型计算分类选择概率;构建价值网络模型,使用所述价值网络模型对所述策略网络模型进行更新;将小批量图像识别场景建模为一个马尔可夫决策过程MDP,通过分类器Agent对所述图像特征以更新后的策略网络模型进行计算,得到分类标签预测。4.根据权利要求3所述的一种小批量图像识别方法,其特征在于,构建策略网络模型,使用所述策略网络模型计算分类选择概率包括如下步骤:采用深度神经网络学习策略函数构建策略网络模型,使用所述策略网络模型对所述策略函数进行多轮次计算,得到多个策略梯度;将多个策略梯度取平均数作为分类选择概率。5.根据权利要求4所述的一种小批量图像识别方法,其特征在于,使用所述策略网络模型对所述策略函数进行多轮次计算之前还包括:构建梯度函数,采用所述梯度函数对策略函数中的权重进行更新,其中,所述梯度函数表示如下:
▽
θ
J
θ
(s)=E
πθ
[
▽
θ
lnπ
θ
(a
t
|s
t
)[Q
π
(s
t
,a
t
)
‑
V<...
【专利技术属性】
技术研发人员:杨娟,肖朝清,郑艺泽,沈非凡,刘磊,
申请(专利权)人:中国人民武装警察部队警官学院,
类型:发明
国别省市:
还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。