模型训练方法、分蘖数预测方法、装置、设备及介质与流程

本发明涉及图像处理领域，具体地，涉及一种模型训练方法、分蘖数预测方法、装置、设备及介质。

背景技术：

1、随着对水稻等农作物的分蘖数与产量的关系研究发现，农作物的分蘖包括有效分蘖和无效分蘖。农作物的有效分蘖数越高，则农作物的产量越高。农作物的无效分蘖增多，不仅无法提高农作物的产量，还将导致农作物的养分浪费。此外，农作物的分蘖数过多，还将导致农作物区域内的农作物密度过大，导致病虫灾害及农作物倒伏。为了提高农作物的产量，需要实时确定农作物的分蘖数，并对农作物的分蘖数进行控制。然而，农作物区域通常种植了大量的农作物，若人工确定每株农作物的分蘖数，需要耗费大量的人力和物力。

2、现有技术中通常利用多光谱设备及高光谱设备等光谱设备观测，技术人员将观测到的农作物光谱数据进行分析处理，得到农作物分蘖数。在确定农作物的分蘖数过程中，需要专门的技术人员按照光谱设备，并对观测到农作物光谱数据进行分析处理，导致确定农作物分蘖数的效率低下。此外，由于需要确定大量农作物区域内农作物分蘖数，导致技术人员需要同时分析处理大量的农作物光谱数据，进一步导致了确定农作物分蘖数的效率低下。

技术实现思路

1、本发明实施例的目的是提供一种设备，该设备用于解决确定农作物分蘖数的效率低下问题。

2、为了实现上述目的，第一方面，本技术提供一种模型训练方法，模型训练方法包括：

3、利用图像采集设备对农作物区域进行图像采集，得到农作物样本图像；

4、基于农作物区域中的农作物的实际分蘖数，对农作物样本图像进行标注，构建训练集；

5、根据swin-transfomer网络结构，构建预设模型；

6、将训练集输入至预设模型，利用损失函数训练预设模型，得到分蘖数确定模型。

7、结合第一方面，在第一种可能的实现方式中，根据swin-transfomer网络结构，构建预设模型，包括：

8、将swin-transfomer网络结构的中间模块的第一子模块与第三子模块连接，并将swin-transfomer网络结构的中间模块的第二子模块与第四子模块连接；

9、将第一子模块、第二子模块、第三子模块及第四子模块的卷积层均更新为可分离卷积层；

10、将swin-transfomer网络结构的输出模块的全连接层更新为预设数量的1x1全卷积层，得到预设模型。

11、结合第一方面，在第二种可能的实现方式中，损失函数为：

12、

13、其中，lδ(y，f(x))为损失函数，δ为损失函数的超参数，y为农作物的实际分蘖数，f(x)为预设模型输出的农作物分蘖数。

14、结合第一方面，在第三种可能的实现方式中，利用多个图像采集设备对农作物区域进行图像采集，得到农作物样本图像，包括：

15、利用多个图像采集设备间隔地对农作物区域进行图像采集，得到农作物样本图像，其中，每个图像采集设备垂直设置于农作物区域的预设高度位置，且每个图像采集设备与农作物区域的边界距离大于或等于预设距离。

16、结合第一方面，在第四种可能的实现方式中，农作物样本图像包括正样本图像和负样本图像，正样本图像为处于种植阶段的农作物图像，负样本图像为处于非种植阶段的农作物图像，基于农作物区域中的农作物的实际分蘖数，对农作物样本图像进行标注，构建训练集，包括：

17、基于农作物区域中的农作物的实际分蘖数，构建分蘖数变化曲线；

18、根据分蘖数变化曲线，对每个正样本图像进行标注；

19、基于预设分蘖数，对每个负样本图像进行标注；

20、将标注后的负样本图像与标注后的正样本图像进行合并，构建训练集。

21、结合第一方面的第四种可能的实现方式，在第五种可能的实现方式中，基于农作物区域中的农作物的实际分蘖数，构建分蘖数变化曲线，包括：

22、获取图像采集设备在农作物区域中的图像采集区域，并将农作物区域中除图像采集区域以外的区域确定为非图像采集区域；

23、基于预设时间间隔对非图像采集区域进行农作物的分蘖数采样，得到至少两组农作物的实际分蘖数；

24、将时间作为第一坐标，将农作物分蘖数作为第二坐标，并将每组农作物的实际分蘖数作为点，构建分蘖数变化曲线。

25、结合第一方面，在第六种可能的实现方式中，模型训练方法还包括：

26、根据农作物区域的农作物种植密度，对农作物样本图像进行截取处理。

27、第二方面，本技术提供一种分蘖数预测方法，分蘖数预测方法包括：

28、利用图像采集设备对目标农作物区域进行图像采集，得到目标农作物图像；

29、将目标农作物图像输入至分蘖数确定模型，得到目标农作物区域的农作物分蘖数，其中，分蘖数确定模型根据如第一方面的模型训练方法得到。

30、第三方面，本技术提供一种模型训练装置，模型训练装置包括：

31、样本图像得到模块，用于利用图像采集设备对农作物区域进行图像采集，得到农作物样本图像；

32、训练集构建模块，用于基于农作物区域中的农作物的实际分蘖数，对农作物样本图像进行标注，构建训练集；

33、预设模型构建模块，用于根据swin-transfomer网络结构，构建预设模型；

34、预设模型训练模块，用于将训练集输入至预设模型，利用损失函数训练预设模型，得到分蘖数确定模型。

35、第四方面，本技术提供一种分蘖数预测装置，分蘖数预测装置包括：

36、目标图像得到模块，用于利用图像采集设备对目标农作物区域进行图像采集，得到目标农作物图像；

37、分蘖数得到模块，用于将目标农作物图像输入至分蘖数确定模型，得到目标农作物区域的农作物分蘖数，其中，分蘖数确定模型根据如第一方面的模型训练方法得到。

38、第五方面，本技术提供一种计算机设备，计算机设备包括存储器及处理器，存储器存储有计算机程序，计算机程序在处理器执行时，实现如第一方面的模型训练方法，或实现如第二方面的分蘖数预测方法。

39、第六方面，本技术提供一种机器可读存储介质，机器可读存储介质上存储有计算机程序，计算机程序被处理器执行时，实现如第一方面的模型训练方法，或实现如第二方面的分蘖数预测方法。

40、本技术提供一种模型训练方法，模型训练方法包括：利用图像采集设备对农作物区域进行图像采集，得到农作物样本图像；基于农作物区域中的农作物的实际分蘖数，对农作物样本图像进行标注，构建训练集；根据swin-transfomer网络结构，构建预设模型；将训练集输入至预设模型，利用损失函数训练预设模型，得到分蘖数确定模型。通过训练基于swin-transfomer网络结构的预设模型，得到分蘖数确定模型。利用分蘖数确定模型可以高效准确得到农作物区域中的农作物分蘖数，不再需要人工确定农作物分蘖数，提高了确定农作物分蘖数的效率。