一种小批量图像识别方法、设备及介质

本发明涉及图像识别，具体涉及一种小批量图像识别方法、设备及介质。

背景技术：

1、某些特殊领域(如战场、罕见病例医疗图像等)的图像由于难以获得、不易标注、重复性低等特点，只能获得小批量图像，导致可供利用的图像特征数据很少，从而增加了图像识别的难度。

2、现有技术通常采用深度学习对于图像进行识别，但是采用深度学习对于图像进行识别通常依赖于大量高质量的人工标注数据，在小批量图像识别任务中，由于显而易见的数据标注的困难性及数据本身的稀缺性，大量有标注图像数据难以获取。

3、其次，特殊领域的新技术装备不断更新，新的特殊领域目标也会不断涌现。而，由于新的特殊领域目标图像所采集到的数据存在不完整性和不确定性，导致可供利用的图像特征数据很少，采用深度学习难以对其进行识别。

4、因此，如何对小批量图像进行高精度识别成为研究热点。

技术实现思路

1、本发明所要解决的技术问题是由于小批量图像所采集到的数据存在不完整性和不确定性，导致难以对小批量图像进行精准识别，目的在于提供一种小批量图像识别方法、设备及介质，通过将采集到的为数不多的特殊领域目标图像作为原始图像输入，经过深度卷积神经网络提取特征，送入策略网络模型和价值网络模型加以推理决策，以实现端到端的小批量图像识别学习，解决了由于小批量图像所采集到的数据存在不完整性和不确定性，导致难以对小批量图像进行精准识别的问题。

2、本发明通过下述技术方案实现：

3、本发明第一方面提供了一种小批量图像识别方法，包括如下步骤：

4、s1、获取小批量图像；

5、s2、构建特征提取网络模型，采用所述特征提取网络模型对所述小批量图像进行特征提取，得到所述小批量图像的图像特征；

6、s3、构建小批量图像分类模型，将所述图像特征送入所述小批量图像分类模型中进行预测，得到分类标签预测；

7、s4、根据分类标签预测对所述特征提取网络模型和所述小批量图像分类模型进行训练，得到小批量图像识别模型sdrl，采用所述小批量图像识别模型sdrl对所述小批量图像进行识别，得到目标图像。

8、在上述技术方案中，首先使用特征提取网络模型对小批量图像进行特征提取得到小批量图像的图像特征，再使用小批量图像分类模型对图像特征进行推理与决策从而完成对小批量图像的图像特征预测，通过分类预测可以提取出小批量图像中的关键图像特征，即使小批量图像所采集的数据不完整也可以得出小批量图像的关键图像特征。

9、以预测后的结果对特征提取网络模型和小批量图像分类模型进行训练最终构建出小批量图像识别模型sdrl，采用小批量图像识别模型sdrl对小批量图像进行识别，得到目标图像。

10、采用强化学习的方法，在只输入原始图像的情况下，通过交互学习的方式，实现信息不完备和不确定情况下的小批量图像识别。充分利用深度学习的特征提取及强化学习的智能决策功能，构建出高精度的特殊领域小批量图像识别模型sdrl。通过小批量图像识别模型sdrl可以解决某些特殊领域中仅靠有限数据量及有限计算资源，实现边缘计算图像识别的问题，最终可实现快速精准目标识别的效果。

11、在一种可选实施例中，构建特征提取网络模型，采用所述特征提取网络模型对所述小批量图像进行特征提取，得到所述小批量图像的图像特征包括如下步骤：

12、构建11层卷积神经网络m_net；其中，所述卷积神经网络的前10层由卷积层和最大池化层交替构成，所述卷积神经网络m_net的第11层由正则化层dropout构成；

13、采用所述卷积层提取所述小批量图像的图像信息并通过relu函数进行激活后传递给所述最大池化层；

14、采用所述最大池化层对所述图像信息进行挑选；

15、采用所述正则化层dropout随机删除所述卷积神经网络m_net中的部分神经元。

16、在一种可选实施例中，构建小批量图像分类模型，将所述图像特征送入所述小批量图像分类模型中进行预测，得到分类标签预测包括如下步骤：

17、构建策略网络模型，使用所述策略网络模型计算分类选择概率；

18、构建价值网络模型，使用所述价值网络模型对所述策略网络模型进行更新；

19、将小批量图像识别场景建模为一个马尔可夫决策过程mdp，通过分类器agent对所述图像特征以更新后的策略网络模型进行计算，得到分类标签预测。

20、在一种可选实施例中，构建策略网络模型，使用所述策略网络模型计算分类选择概率包括如下步骤：

21、采用深度神经网络学习策略函数构建策略网络模型，使用所述策略网络模型对所述策略函数进行多轮次计算，得到多个策略梯度；

22、将多个策略梯度取平均数作为分类选择概率。

23、在一种可选实施例中，使用所述策略网络模型对所述策略函数进行多轮次计算之前还包括：

24、构建梯度函数，采用所述梯度函数对策略函数中的权重进行更新，其中，所述梯度函数表示如下：

25、

26、其中，为梯度函数，π(a/s,θ)为策略函数，vπ(st)为状态st拥有的基准值，qπ(st,at)为状态st下由策略π选择动作at的期望累计回报动作值函数。

27、在一种可选实施例中，构建梯度函数之后还包括：

28、通过所述期望累计回报动作值函数和状态st拥有的基准值构建优势函数，所述优势函数用于降低因状态基准值的变化引起的方差；

29、其中，所述优势函数表示如下：

30、aπ(st,at)＝qπ(st,at)-vπ(st)

31、其中，aπ(st,at)为优势函数。

32、在一种可选实施例中，构建价值网络模型，使用所述价值网络模型对所述策略网络模型进行更新包括如下步骤：

33、采用深度神经网络学习价值函数构建价值网络模型；

34、将策略梯度输入所述价值网络模型中，所述价值网络模型根据所述策略梯度进行更新；

35、将更新后的价值网络模型反馈至所述策略网络模型中，对所述策略网络模型进行更新。

36、在一种可选实施例中，将策略梯度输入所述价值网络模型中，所述价值网络模型根据所述策略梯度进行更新之后还包括：

37、使用所述价值网络模型对所述价值函数进行多轮次计算，得到多个价值梯度；

38、将所述价值梯度和所述策略梯度输入所述价值网络模型中对所述价值梯度进行更新，得到更新后的价值梯度；

39、将多个更新后的价值梯度取平均值，得到目标价值参数；

40、将所述目标价值参数反馈至所述价值网络模型中，对所述价值网络模型进行更新。

41、本发明第二方面提供了一种电子设备，包括存储器、处理器及存储在存储器上并可在处理器上运行的计算机程序，所述处理器执行所述程序时实现一种小批量图像识别方法。

42、本发明第三方面提供了一种计算机可读存储介质，其上存储有计算机程序，该程序被处理器执行时实现一种小批量图像识别方法。

43、本发明与现有技术相比，具有如下的优点和有益效果：

44、本发明通过将采集到的为数不多的特殊领域目标图像作为原始图像输入，经过深度卷积神经网络提取特征，送入策略网络模型和价值网络模型加以推理决策，以实现端到端的小批量图像识别学习，解决了由于小批量图像所采集到的数据存在不完整性和不确定性，导致难以对小批量图像进行精准识别的问题。