分类模型训练和图像特征提取方法及装置与流程

文档序号:31079087发布日期:2022-08-09 22:06阅读:218来源:国知局
分类模型训练和图像特征提取方法及装置与流程

1.本技术涉及图像技术领域,尤其涉及一种分类模型训练和图像特征提取方法及装置。


背景技术:

2.随着图像处理技术的快速发展,图像特征提取技术越发成熟,图像特征数据被广泛应用于各种图像处理任务中,如三维地图重建、目标检测和类型识别等。图像特征提取是图像处理任务中的重要环节之一,有效特征数据是图像处理任务成功的关键。
3.目前提取图像中的有效特征数据时,不对图像进行分类,而是直接通过深度学习模型从图像中提取特征数据。当图像数据集覆盖的数据类型较广泛时,该深度学习模型只能提取到特定类型图像中的有效特征数据。而其余图像的特征数据并不能准确表达图像中的信息,影响特征数据的提取结果,降低图像处理任务的成功率。因此直接通过深度学习模型进行特征提取的方案不适用于复杂场景图像的特征提取。


技术实现要素:

4.本技术提供一种分类模型训练和图像特征提取方法及装置,解决了现有技术不适用于提取复杂场景图像的特征的问题,提高特征提取方法的适用性,高效提取各种复杂场景图像的有效特征数据,提高图像处理任务的成功率。
5.第一方面,本技术提供了一种分类模型训练方法,包括:
6.依次获取最新的训练样本集合,通过每次获取到的训练样本集合中的样本图像训练出对应的第一特征提取模型;
7.根据每次训练出的第一特征提取模型输出的对应样本图像的描述值,从当前训练样本集合中筛选出目标样本集合并对所述训练样本集合进行更新,确定每次筛选出的目标样本集合对应的目标特征提取模型,所述目标特征提取模型为第一特征提取模型或第二特征提取模型,所述第二特征提取模型由所述目标样本集合训练得到;
8.基于多次训练得到的目标特征提取模型和对应的目标样本集合,训练得到数据分类模型。
9.第二方面,本技术提供了一种图像特征提取方法,包括:
10.获取待提取特征的图像数据,将所述图像数据输入预设的数据分类模型中,得到所述数据分类模型输出的分类结果;其中,所述数据分类模型通过第一方面所述的分类模型训练方法训练得到;
11.根据所述分类结果确定所述图像数据对应的目标样本集合,将所述目标样本集合对应的目标特征提取模型确定为所述图像数据对应的目标特征提取模型;其中,所述目标特征提取模型通过第一方面所述的分类模型训练方法训练得到;
12.将所述图像数据输入对应的目标特征提取模型中,得到所述目标特征提取模型输出的所述图像数据的特征信息。
13.第三方面,本技术提供了一种分类模型训练装置,包括:
14.第一训练模块,被配置为依次获取最新的训练样本集合,通过每次获取到的训练样本集合中的样本图像训练出对应的第一特征提取模型;
15.第二训练模块,被配置为根据每次训练出的第一特征提取模型输出的对应样本图像的描述值,从当前训练样本集合中筛选出目标样本集合并对所述训练样本集合进行更新,确定每次筛选出的目标样本集合对应的目标特征提取模型,所述目标特征提取模型为第一特征提取模型或第二特征提取模型,所述第二特征提取模型由所述目标样本集合训练得到;
16.第三训练模块,被配置为基于多次训练得到的目标特征提取模型和对应的目标样本集合,训练得到数据分类模型。
17.第四方面,本技术提供了一种图像特征提取装置,包括:
18.数据分类模块,被配置为获取待提取特征的图像数据,将所述图像数据输入预设的数据分类模型中,得到所述数据分类模型输出的分类结果;其中,所述数据分类模型通过第一方面所述的分类模型训练方法训练得到;
19.第一模型确定模块,被配置为根据所述分类结果确定所述图像数据对应的目标样本集合,将所述目标样本集合对应的目标特征提取模型确定为所述图像数据对应的目标特征提取模型;其中,所述目标特征提取模型通过第一方面所述的分类模型训练方法训练得到;
20.第一特征提取模块,被配置为将所述图像数据输入对应的目标特征提取模型中,得到所述目标特征提取模型输出的所述图像数据的特征信息。
21.第五方面,本技术提供了一种电子设备,包括:
22.一个或多个处理器;存储装置,存储一个或多个程序,当所述一个或多个程序被所述一个或多个处理器执行,使得所述一个或多个处理器实现如第一方面所述的分类模型训练方法或如第二方面所述的图像特征提取方法。
23.第六方面,本技术提供了一种包含计算机可执行指令的存储介质,所述计算机可执行指令在由计算机处理器执行时用于执行如第一方面的分类模型训练方法或第二方面所述的图像特征提取方法。
24.本技术通过第一特征提取模型输出的描述值,将第一特征提取模型可以提取到有效特征信息的样本图像划分至目标样本集合中,目标样本集合中的样本图像中的有效特征信息分布于第一特征提取模型重点关注的分布区域,因此第一特征提取模型可作为目标样本集合对应的目标特征提取模型,以用于重点提取目标样本集合中样本图像的有效特征信息所在的分布区域的特征信息。第二特征提取模型是基于目标样本集合训练得到的,其着重训练对目标样本集合中样本图像的有效特征信息所在分布区域进行特征提取,模型更加收敛。对于训练样本集合筛选出目标样本集合后剩余的样本图像,这部分样本图像的有效特征信息分布在其他分布区域,在后续目标特征提取模型训练过程中,可训练出重点关注其他分布区域的目标特征提取模型,实现多种目标特征提取模型的训练,以便后续通过多种目标特征提取模型从各种类型的图像数据中提取到有效特征信息。目标特征提取模型对应的目标样本集合中的样本图像属于同一数据类型,因此可将样本图像所属的目标样本集合作为该样本图像的标签信息,通过多个目标样本集合训练出数据分类模型,以便后续根
据数据分类模型确定与图像数据的特征信息类型相类似的目标样本集合,并通过目标样本集合对应的目标特征提取模型提取到图像数据中的有效特征信息,保证特征信息的可靠性,提高图像处理任务的成功率。通过多种目标特征提取模型可高效提取各种复杂场景图像的有效特征数据,提高特征提取方法的适用性。
附图说明
25.图1是本技术实施例提供的一种分类模型训练方法的流程图;
26.图2是本技术实施例提供的通过训练样本集合训练目标特征提取模型的流程图;
27.图3是本技术实施例提供的划分每一批次的训练样本集合的流程图;
28.图4是本技术实施例提供的通过目标样本集合训练目标特征提取模型的流程图;
29.图5是本技术实施例提供的第一特征提取模型和第二特征提取模型的损失值的示意图;
30.图6是本技术实施例提供的数据分类模型训练阶段的流程图;
31.图7是本技术实施例提供的一种图像特征提取方法的流程图;
32.图8是本技术实施例提供的提取图像序列的特征信息的流程图;
33.图9为本技术实施例提供的一种分类模型训练装置的结构示意图;
34.图10为本技术实施例提供的一种图像特征提取装置的结构示意图;
35.图11是本技术实施例提供的一种电子设备的结构示意图。
具体实施方式
36.为了使本技术的目的、技术方案和优点更加清楚,下面结合附图对本技术具体实施例作进一步的详细描述。可以理解的是,此处所描述的具体实施例仅仅用于解释本技术,而非对本技术的限定。另外还需要说明的是,为了便于描述,附图中仅示出了与本技术相关的部分而非全部内容。在更加详细地讨论示例性实施例之前应当提到的是,一些示例性实施例被描述成作为流程图描绘的处理或方法。虽然流程图将各项操作(或步骤)描述成顺序的处理,但是其中的许多操作可以被并行地、并发地或者同时实施。此外,各项操作的顺序可以被重新安排。当其操作完成时所述处理可以被终止,但是还可以具有未包括在附图中的附加步骤。所述处理可以对应于方法、函数、规程、子例程、子程序等等。
37.本技术的说明书和权利要求书中的术语“第一”、“第二”等是用于区别类似的对象,而不用于描述特定的顺序或先后次序。应该理解这样使用的数据在适当情况下可以互换,以便本技术的实施例能够以除了在这里图示或描述的那些以外的顺序实施,且“第一”、“第二”等所区分的对象通常为一类,并不限定对象的个数,例如第一对象可以是一个,也可以是多个。此外,说明书以及权利要求中“和/或”表示所连接对象的至少其中之一,字符“/”,一般表示前后关联对象是一种“或”的关系。
38.本实施例中提供的分类模型训练方法可以由分类模型训练设备执行,图像特征提取方法可以由图像特征提取设备执行,分类模型训练设备和图像特征提取设备可以是同一电子设备,也可以是不同的电子设备。该电子设备可以通过软件和/或硬件的方式实现,该电子设备可以是两个或多个物理实体构成,也可以是一个物理实体构成。例如该电子设备可以是服务器等运算能力较强的设备,也可以是采集图像数据的智能设备。
39.该电子设备安装有至少一类操作系统,其中,操作系统包括但不限定于安卓系统、linux系统及windows系统。该电子设备可以基于操作系统安装至少一个应用程序,应用程序可以为操作系统自带的应用程序,也可以为从第三方设备或者服务器中下载的应用程序,实施例中,该电子设备至少按照有可以执行分类模型训练方法和/或图像特征提取方法的应用程序,因此,电子设备也可以是应用程序本身。
40.为便于理解,本实施例以服务器为执行分类模型训练方法和图像特征提取方法的主体为例,进行描述。
41.在一实施例中,以图像处理任务为测绘区域的三维地图重建为例进行描述。示例性的,无人设备采集测绘区域的遥感影像数据,并根据遥感影像数据的特征信息构建测绘区域的三维地图为例进行描述。遥感影像数据中的特征信息的精度直接影响到三维地图的精度。传统图像特征提取方法是通过深度学习模型提取遥感影像数据中的特征数据。由于不同类型的图像数据中的有效特征区域不同,使用深度学习模型这类只能提取到单一类型的特征数据的特征提取方法时,会重点关注对应类型中的有效特征区域,导致遥感影像数据中只有部分图像的特征数据能够有效表达出对应图像中的图像信息。而其余图像的特征数据因深度学习模型重点关注的特征区域包括无关或不重要信息,导致该特征数据不能准确表达对应图像中的图像信息。因此当测绘区域的地域跨度较大,图像数据集覆盖的数据类型较广泛时,传统图像特征提取方法存在特征数据的提取结果差等问题,降低了图像处理任务的成功率。
42.为解决上述问题,本实施例提供了一种分类模型训练方法和图像特征提取方法,以优化特征提取的提取结果,提高图像处理任务的成功率。
43.图1给出了本技术实施例提供的一种分类模型训练方法的流程图。如图1所示,该分类模型训练方法具体包括:
44.s110、依次获取最新的训练样本集合,通过每次获取到的训练样本集合中的样本图像训练出对应的第一特征提取模型,根据每次训练出的第一特征提取模型输出的对应样本图像的描述值,从当前训练样本集合中筛选出目标样本集合并对训练样本集合进行更新,确定每次筛选出的目标样本集合对应的目标特征提取模型,目标特征提取模型为第一特征提取模型或第二特征提取模型,第二特征提取模型由目标样本集合训练得到。
45.其中,训练样本集合包括多个样本图像,样本图像为训练阶段用于训练目标特征提取模型的图像数据。示例性的,无人设备可事先通过多个飞行架次采集的遥感图像数据,并将该遥感图像数据作为样本图像。在本实施例中,通过分批训练多个目标特征提取模型,每一批次对应一个训练样本集合,基于每一批次的训练样本集合训练出一个目标特征提取模型。其中,下一批次的训练样本集合对应从上一批次的训练样本集合中获取,因此每一批次训练时都会获取到新的训练样本集合。
46.在一实施例中,事先获取总训练样本集合,通过总训练样本集合进行第一批次训练后得到第一个目标特征提取模型,从总训练样本集合中筛选出目标样本集合,将总训练样本集合中剩余的样本图像更新为第二批次的训练样本集合,以通过更新后的训练样本集合训练出第二个目标特征提取模型,依次类推,直至目标特征提取模型的数量等于预设数量。其中,目标样本集合可看作包括同一数据类型的样本图像的集合,目标特征提取模型为一种神经网络模型,其用于重点关注对应目标样本集合中的样本图像的有效特征信息的分
布区域。可理解,每一批次用于训练目标特征提取模型的样本图像是不同的,每一批次对应训练出一种图像数据类型对应的目标特征提取模型,多个批次后可训练出多种图像数据类型对应的目标特征提取模型。相应的,在图像特征提取过程中,如果待提取特征的图像数据的数据类型与某一目标样本集合中的样本图像的数据类型相同,即图像数据中有效特征信息的分布区域与同一类型的样本图像中的有效特征信息的分布区域相同。那么通过对应的目标样本集合的目标特征提取模型提取图像数据中的特征信息时,目标特征提取模型会重点关注其有效特征信息所在的区域,进而目标特征提取模型会提取到该图像数据的有效特征信息。
47.本实施例提供了两种生成目标特征提取模型的实现方式,第一种是将训练样本集合训练出的第一特征提取模型作为目标特征提取模型,第二种是将目标样本集合训练出的第二特征提取模型作为目标特征提取模型。通过以下实施例分别描述两种实现方式的具体步骤。
48.在将训练样本集合训练出的第一特征提取模型作为目标特征提取模型的实施例中,图2是本技术实施例提供的通过训练样本集合训练目标特征提取模型的流程图。如图2所示,通过训练样本集合训练目标特征提取模型的步骤具体包括s1101-s1105:
49.s1101、通过当前训练样本集合中的样本图像对预设的第一神经网络模型进行训练,得到当前第一特征提取模型。
50.其中,第一神经网络模型由卷积网络构成,卷积网络可用于提取图像数据中深度特征。在每一批次通过对应的训练样本集合对同一第一神经网络模型进行训练,在网络模型结构不变的情况下,通过网络模型参数的差异性反应每一批次的第一特征提取模型的差异性。通过第一特征提取模型对应的模型参数规定模型对图像各区域的关注度,进而体现出不同批次的第一特征提取模型提取到的特征信息的差异性。
51.示例性的,在第一批次的训练阶段,将总训练样本集合作为第一批次的训练样本集合d1。对训练样本集合d1中的每一样本图像进行初始化,如赋予每一样本图像同一初始权重值,并通过初始化后的样本图像训练第一神经网络模型。其中,将初始化后的样本图像输入第一神经网络模型,得到第一神经网络模型输出的特征信息和描述值,该描述值可以是损失值或置信度值等。根据第一神经网络模型输出的特征信息和描述值调整模型参数,以将第一神经网络模型训练成重点关注描述值较高的样本图像的特征信息的分布区域,得到第一批次的第一特征提取模型m1。
52.s1102、将当前训练样本集合中的每一样本图像输入当前第一特征提取模型,得到当前第一特征提取模型输出的对应样本图像的描述值。
53.示例性的,将样本图像输入第一特征提取模型m1,得到第一特征提取模型m1输出的特征信息和描述值。由于第一特征提取模型m1重点关注图像中的某一分布区域的特征信息,当第一特征提取模型m1从重点关注的分布区域中提取到样本图像的有效特征信息时,第一特征提取模型m1会输出较高的描述值。相反的,当第一特征提取模型m1从重点关注的分布区域中提取到样本图像的无效特征信息时,第一特征提取模型m1会输出较低的描述值。因此,通过样本图像的描述值可确定样本图像的有效特征信息是否分布于第一特征提取模型重点关注的分布区域,进而确定样本图像的数据类型是否为第一特征提取模型对应的数据类型。
54.s1103、根据描述值从当前训练样本集合中筛选出当前目标样本集合,并将当前训练样本集合剩余的样本图像更新为下一次训练的训练样本集合,目标样本集合中的样本图像的描述值大于剩余的样本图像的描述值。
55.示例性的,当样本图像的数据类型为第一特征提取模型m1对应的数据类型时,可将样本图像划分至第一批次的目标样本集合d1’中,以便后续通过目标样本集合和对应的目标特征提取模型训练数据分类模型。当样本图像的数据类型不为第一特征提取模型m1对应的数据类型时,可将样本图像划分至第二批次的训练样本集合d2中,以便后续训练出不同数据类型对应的目标特征提取模型。
56.在一实施例中,设置描述值阈值,描述值阈值可理解为每一批次的目标样本集合中的样本图像的最低描述值。当样本图像的描述值大于描述值阈值时,确定该样本图像属于对应批次的目标样本集合,当样本图像的描述值小于或等于描述值阈值时,确定该样本图像不属于对应批次的目标样本集合。示例性的,根据描述值阈值将训练样本集合d1划分为第一批次的目标样本集合d1’和第二批次的训练样本集合d2。在该实施例中,以描述平均值作为描述值阈值划分每一批次的训练样本集合。图3是本技术实施例提供的划分每一批次的训练样本集合的流程图。如图3所示,划分每一批次的训练样本集合的步骤具体包括s11031-s11032:
57.s11031、根据当前训练样本集合中每一样本图像的描述值,计算当前训练样本集合的描述平均值。
58.以描述值为置信度值为例进行描述。将第一批次的训练样本集合d1中每一样本图像的置信度值求和再除以样本图像数量,得到置信度平均值v
avg

59.s11032、将描述值大于描述平均值的样本图像划分至当前目标样本集合,将描述值小于或等于描述平均值的样本图像划分至下一次训练的训练样本集合。
60.示例性的,将置信度值大于v
avg
的样本图像划分至第一批次的目标样本集合d1’,将置信度值小于或等于v
avg
的样本图像划分至第二批次的训练样本集合d2。
61.在另一实施例中,利用高斯分布拟合第一批次的训练样本集合d1中每个样本图像的置信度分布均值mu和方差sigma,计算样本图像的置信度是否在整体分布(mu-0.5*sigma,mu+0.5*sigma)的范围内。如果在则将该样本图像划分至第一批次的目标样本集合d1’,如果不在则将该样本图像划分至第二批次的训练样本集合d2。
62.s1104、将当前第一特征提取模型设置为当前筛选出的目标样本集合对应的目标特征提取模型。
63.示例性的,由于目标样本集合中的样本图像是基于第一特征提取模型输出的描述值进行划分的,即目标样本集合中的样本图像的有效特征信息所在的分布区域,正好是第一特征提取模型重点关注的分布区域。因此可将第一特征提取模型作为目标样本集合对应的目标特征提取模型,以用于重点提取目标样本集合中这类样本图像的有效特征信息所在的分布区域的特征信息,优化各种数据类型的特征提取的效果。
64.s1105、确定第一特征提取模型的数量小于预设数量时,通过下一批次的训练样本集合训练出下一批次的第一特征提取模型,直至第一特征提取模型的数量等于预设数量。
65.假设预设数量为n,n》1。示例性的,在获取到第一批次的第一特征提取模型m1和目标样本集合d1’后,通过第二批次的训练样本集合d2训练出第二批次的第一特征提取模型
m2,并将第二批次训练得到的第一特征提取模型m2作为目标特征提取模型。示例性的,对第二批次的训练样本集合d2中的样本图像赋予同一权重值,并通过权重处理后的样本图像训练第一神经网络模型,得到第二批次的第一特征提取模型m2。将第二批次的训练样本集合d2的样本图像输入第二批次的第一特征提取模型m2中,得到第二批次的第一特征提取模型m2输出的描述值。根据描述值将第二批次的训练样本集合d2划分为第二批次的目标样本集合d2’和第三批次的训练样本集合d3。以此类推,在第i批次的训练阶段时,通过第i批次的训练样本集合di训练出第i批次的第一特征提取模型mi,并将第i批次训练样本集合di输入第i批次第一特征提取模型mi,根据第i批次的第一特征提取模型mi输出的描述值,确定第i批次的目标样本集合di’和第i+1批次的训练样本集合di+1。当i=n时,获取到第n批次的第一特征提取模型mn和目标样本集合d
n’后,停止训练。
66.在将目标样本集合训练出的第二特征提取模型作为目标特征提取模型的实施例中,图4是本技术实施例提供的通过目标样本集合训练目标特征提取模型的流程图。如图4所示,通过目标样本集合训练目标特征提取模型的步骤具体包括s1106-s1110:
67.s1106、通过当前训练样本集合中的样本图像对预设的第一神经网络模型进行训练,得到当前第一特征提取模型。
68.s1107、将当前训练样本集合中的每一样本图像输入当前第一特征提取模型,得到当前第一特征提取模型输出的对应样本图像的描述值。
69.s1108、根据描述值从当前训练样本集合中筛选出当前目标样本集合,并将当前训练样本集合剩余的样本图像更新为下一次训练的训练样本集合,目标样本集合中的样本图像的描述值大于剩余的样本图像的描述值。
70.示例性的,步骤s1106-步骤s1108可参考步骤s1101-s1103。需要说明的,本实施例中的第一特征提取模型是用于筛选目标样本集合的模型,而非目标特征提取模型。
71.s1109、通过当前筛选出的目标样本集合对预设的第一神经网络模型进行训练,得到当前第二特征提取模型,并将当前第二特征提取模型设置为当前筛选出的目标样本集合对应的目标特征提取模型。
72.示例性的,通过第i批次的目标样本集合di’训练第一神经网络模型,得到第i批次的第二特征提取模型mi’,并将第i批次的第二特征提取模型mi’作为第i批次的目标样本集合对应的目标特征提取模型。
73.s1110、确定第二特征提取模型的数量小于预设数量时,从最新生成的训练样本集合中获取下一批次的目标样本集合,并基于下一批次的目标样本集合训练出下一批次的第二特征提取模型,直至第二特征提取模型的数量等于预设数量。
74.示例性的,通过第i批次的训练样本集合di训练出第i批次的第一特征提取模型mi,并将训练样本集合di中的样本图像输入第一特征提取模型mi,得到第一特征提取模型mi输出的描述值。根据该描述值将训练样本集合di划分为第i+1批次的训练样本集合di+1和第i批次的目标样本集合di’。通过第i批次的目标样本集合di’训练出第i批次的第二特征提取模型mi’,并将第二特征提取模型mi’作为第i批次的目标特征提取模型。当i=n时,获取到第n批次的第二特征提取模型m
n’和目标样本集合d
n’后,停止训练。
75.图5是本技术实施例提供的第一特征提取模型和第二特征提取模型的损失值的示意图。如图5所示,在第i批次的目标特征提取模型中,第一特征提取模型xi的损失值li大于
第二特征提取模型xi’的损失值li’。其中,损失值可以表征模型的收敛程度,损失值越大收敛程度越小,损失值越小,收敛程度越大,因此第二特征提取模型xi’比第一特征提取模型xi更加收敛,第二特征提取模型xi’更能描述对应目标样本集合di’的特征信息。
76.s120、基于多次训练得到的目标特征提取模型和对应的目标样本集合,训练得到数据分类模型。
77.其中,数据分类模型为用于确定图像数据的数据类型的神经网络模型。示例性的,数据分类模型包括卷积网络和全连接层,通过卷积网络提取图像数据中的特征信息并将特征信息输入全连接层,通过全连接层根据特征信息输出图像数据的分类结果。
78.在该实施例中,目标特征提取模型对应的目标样本集合中的样本图像属于同一数据类型,因此可将样本图像所属的目标样本集合作为该样本图像的标签信息,通过样本图像和对应的标签信息训练第二神经网络模型,得到数据分类模型。
79.在一实施例中,图6是本技术实施例提供的数据分类模型训练阶段的流程图。如图6所示,数据分类模型训练阶段的步骤具体包括s1201-s1203:
80.s1201、将样本图像所属的目标样本集合作为对应样本图像的标签信息。
81.s1202、将样本图像输入预设的第二神经网络模型,得到第二神经网络模型输出的对应样本图像的分类结果。
82.s1203、根据样本图像对应的分类结果和标签信息,调整第二神经网络模型的参数,得到数据分类模型。
83.本实施例以目标特征提取模型为第二特征提取模型为例进行描述。经过分批训练多个目标特征提取模型后,得到各个目标特征提取模型(m1’,m2’,

,m
n’)对应的目标样本集合{d1’,d2’,

,d
n’}。通过目标样本集合中的样本图像训练出数据分类模型mc。
84.在另一实施例中,在第n批次的训练阶段,还获取到了第n批次除目标样本集合d
n’以外的样本图像,将这些样本图像划分至困难样本集合d
n+1’中。通过目标样本集合{d1’,d2’,

,d
n’}和困难样本集合d
n+1’中的样本图像训练出数据分类模型mc。在数据分类模型训练阶段,将样本图像输入第二神经网络模型,得到第二神经网络模型输出的分类结果。将样本图像的分类结果和对应的标签信息代入预设的损失函数,根据损失函数输出的损失结果调整第二神经网络模型的模型参数。当第二神经网络模型收敛或迭代次数达到上限时,完成训练并获取到数据分类模型mc。
85.在该实施例中,将图像数据输入数据分类模型mc,数据分类模型mc确定该图像数据属于困难样本集合d
n+1’,即图像数据不属于任一目标样本集合。进而确定该图像数据中不存在有效特征信息,因此可将该图像数据作为无效数据,不提取该图像数据中的特征信息。
86.在上述实施例的基础上,将分类模型训练方法训练得到的数据分类模型和目标特征提取模型应用到图像特征提取方法中,以提取到图像数据中的有效特征信息。图7是本技术实施例提供的一种图像特征提取方法的流程图。参考图7,该图像特征提取方法具体包括:
87.s210、获取待提取特征的图像数据,将图像数据输入预设的数据分类模型中,得到数据分类模型输出的分类结果。
88.在本实施例中,数据分类模型通过步骤s110-s120描述的分类训练方法训练得到。
待提取特征的图像数据为无人设备执行测绘任务时采集到的包含有测绘区域的遥感影像数据。无人设备在完成测绘任务后通过蜂窝网络或无线通信将图像数据传输至服务器,服务器接收到图像数据后执行本实施例中的图像特征提取方法,以获取到图像数据中关于测绘区域的特征信息。
89.在一实施例中,数据分类模型输出的分类结果可以为分类数值,如数据分类模型预先设置五种数据类型,每种数据类型对应一个分类数值范围,如[0,1]、(1,2]、(2,3]、(3,4]和(4,5]。根据数据分类模型输出的分类数值所在的分类数值范围,确定图像数据的数据类型。
[0090]
s220、根据分类结果确定图像数据对应的目标样本集合,将目标样本集合对应的目标特征提取模型确定为图像数据对应的目标特征提取模型。
[0091]
在本实施例中,目标特征提取模型通过步骤s110-s120描述的分类训练方法训练得到。
[0092]
在一实施例中,每一目标样本集合对应一个分类数值范围,根据全连接层输出的分类数值所在的分类数值范围,确定图像数据对应的目标样本集合,进而确定图像数据对应的目标特征提取模型。
[0093]
s230、将图像数据输入对应的目标特征提取模型中,得到目标特征提取模型输出的图像数据的特征信息。
[0094]
示例性的,将图像数据输入对应的目标特征提取模型中,通过目标特征提取模型中的卷积网络从图像数据的有效特征信息的分布区域提取深度特征,得到图像数据的有效特征信息。
[0095]
在上述实施例的基础上,图8是本技术实施例提供的提取图像序列的特征信息的流程图。如图8所示,提取图像序列的特征信息的步骤具体包括s240-s260:
[0096]
s240、从待提取特征的架次图像集合中获取多个图像数据,将多个图像数据输入数据分类模型,得到数据分类模型输出的分类结果。
[0097]
其中,架次图像集合是指无人设备在一个架次中采集到的遥感影像数据。示例性的,如果无人设备在一个架次中采集到的图像数据的内容大致相同,那么可从该架次图像集合中挑选出多个图像数据,根据多个图像数据确定该架次图像集合的数据类型,将该数据类型对应的目标特征提取模型提取该架次中的所有图像数据的特征信息。将多个图像数据输入数据分类模型mc,得到数据分类模型mc输出的分类数值。其中,如果数据分类模型输出的分数数值对应到困难样本集合,则将该图像数据丢弃。
[0098]
s250、对多个图像数据的分类结果进行加权处理,得到架次图像集合的分类结果,根据架次图像集合的分类结果确定架次图像集合对应的目标特征提取模型。
[0099]
示例性的,将多个图像数据的分数数值做加权融合,将加权融合得到的分数数值作为架次图像集合的分类数值。根据该分类数值对应的分数数值范围,确定该分数数值范围对应的目标样本集合中的样本图像与该架次图像集合中的图像数据属于同一数据类型,并确定该目标样本集合对应的目标特征提取模型为对应架次的目标特征提取模型。
[0100]
s260、将架次图像集合中的每一图像数据输入对应的目标特征提取模型中,得到目标特征提取模型输出的特征信息。
[0101]
示例性的,将架次图像集合中的所有图像数据输入对应的目标特征提取模型中,
得到目标特征提取模型输出的每一图像数据的特征信息。通过本实施例无需确定每一图像数据的数据类型,提高特征信息的提取效率。
[0102]
综上,本技术实施例提供的分类模型训练方法和图像特征提取方法,通过第一特征提取模型输出的描述值,将第一特征提取模型可以提取到有效特征信息的样本图像划分至目标样本集合中,目标样本集合中的样本图像中的有效特征信息分布于第一特征提取模型重点关注的分布区域,因此第一特征提取模型可作为目标样本集合对应的目标特征提取模型,以用于重点提取目标样本集合中样本图像的有效特征信息所在的分布区域的特征信息。第二特征提取模型是基于目标样本集合训练得到的,其着重训练对目标样本集合中样本图像的有效特征信息所在分布区域进行特征提取,模型更加收敛。对于训练样本集合筛选出目标样本集合后剩余的样本图像,这部分样本图像的有效特征信息分布在其他分布区域,在后续目标特征提取模型训练过程中,可训练出重点关注其他分布区域的目标特征提取模型,实现多种目标特征提取模型的训练,以便后续通过多种目标特征提取模型从各种类型的图像数据中提取到有效特征信息。目标特征提取模型对应的目标样本集合中的样本图像属于同一数据类型,因此可将样本图像所属的目标样本集合作为该样本图像的标签信息,通过多个目标样本集合训练出数据分类模型,以便后续根据数据分类模型确定与图像数据的特征信息类型相类似的目标样本集合,并通过目标样本集合对应的目标特征提取模型提取到图像数据中的有效特征信息,保证特征信息的可靠性,提高图像处理任务的成功率。通过多种目标特征提取模型可高效提取各种复杂场景图像的有效特征数据,提高特征提取方法的适用性。
[0103]
在上述实施例的基础上,图9为本技术实施例提供的一种分类模型训练装置的结构示意图。参考图9,本实施例提供的分类模型训练装置具体包括:第一训练模块31、第二训练模块32和第三训练模块33:
[0104]
第一训练模块,被配置为依次获取最新的训练样本集合,通过每次获取到的训练样本集合中的样本图像训练出对应的第一特征提取模型;
[0105]
第二训练模块,被配置为根据每次训练出的第一特征提取模型输出的对应样本图像的描述值,从当前训练样本集合中筛选出目标样本集合并对训练样本集合进行更新,确定每次筛选出的目标样本集合对应的目标特征提取模型,目标特征提取模型为第一特征提取模型或第二特征提取模型,第二特征提取模型由目标样本集合训练得到;
[0106]
第三训练模块,被配置为基于多次训练得到的目标特征提取模型和对应的目标样本集合,训练得到数据分类模型。
[0107]
在上述实施例的基础上,第一训练模块包括:第一模型训练单元,被配置为通过当前训练样本集合中的样本图像对预设的第一神经网络模型进行训练,得到当前第一特征提取模型。
[0108]
在上述实施例的基础上,第二训练模块包括:描述值确定单元,被配置为将当前训练样本集合中的每一样本图像输入当前第一特征提取模型,得到当前第一特征提取模型输出的对应样本图像的描述值;目标样本集合确定单元,被配置为根据描述值从当前训练样本集合中筛选出当前目标样本集合,并将当前训练样本集合剩余的样本图像更新为下一次训练的训练样本集合,目标样本集合中的样本图像的描述值大于剩余的样本图像的描述值。
[0109]
在上述实施例的基础上,目标样本集合确定单元包括:描述平均值确定子单元,被配置为根据当前训练样本集合中每一样本图像的描述值,计算当前训练样本集合的描述平均值;样本图像划分子单元,被配置为将描述值大于描述平均值的样本图像划分至当前目标样本集合,将描述值小于或等于描述平均值的样本图像划分至下一次训练的训练样本集合。
[0110]
在上述实施例的基础上,第二训练模块包括:第一模型设置单元,被配置为将当前第一特征提取模型设置为当前筛选出的目标样本集合对应的目标特征提取模型。
[0111]
在上述实施例的基础上,第二训练模块包括:第二模型设置单元,被配置为通过当前筛选出的目标样本集合对预设的第一神经网络模型进行训练,得到当前第二特征提取模型,并将当前第二特征提取模型设置为当前筛选出的目标样本集合对应的目标特征提取模型。
[0112]
在上述实施例的基础上,第三训练模块包括:标签信息确定单元,被配置为将样本图像所属的目标样本集合作为对应样本图像的标签信息;模型分类单元,被配置为将样本图像输入预设的第二神经网络模型,得到第二神经网络模型输出的对应样本图像的分类结果;参数调整单元,被配置为根据样本图像对应的分类结果和标签信息,调整第二神经网络模型的参数,得到数据分类模型。
[0113]
在上述实施例的基础上,图10为本技术实施例提供的一种图像特征提取装置的结构示意图。参考图10,本实施例提供的图像特征提取装置具体包括:数据分类模块41、第一模型确定模块42和第一特征提取模块43:
[0114]
数据分类模块,被配置为获取待提取特征的图像数据,将图像数据输入预设的数据分类模型中,得到数据分类模型输出的分类结果;其中,数据分类模型通过上述的分类模型训练方法训练得到;
[0115]
第一模型确定模块,被配置为根据分类结果确定图像数据对应的目标样本集合,将目标样本集合对应的目标特征提取模型确定为图像数据对应的目标特征提取模型;其中,目标特征提取模型通过上述的分类模型训练方法训练得到;
[0116]
第一特征提取模块,被配置为将图像数据输入对应的目标特征提取模型中,得到目标特征提取模型输出的图像数据的特征信息。
[0117]
在上述实施例的基础上,图像特征提取装置还包括:多图获取模块,被配置为从待提取特征的架次图像集合中获取多个图像数据,将多个图像数据输入数据分类模型,得到数据分类模型输出的分类结果;第二模型确定模块,被配置为对多个图像数据的分类结果进行加权处理,得到架次图像集合的分类结果,根据架次图像集合的分类结果确定架次图像集合对应的目标特征提取模型;第二特征提取模块,被配置为将架次图像集合中的每一图像数据输入对应的目标特征提取模型中,得到目标特征提取模型输出的特征信息。
[0118]
上述,本技术实施例提供的分类模型训练装置和图像特征提取装置,通过第一特征提取模型输出的描述值,将第一特征提取模型可以提取到有效特征信息的样本图像划分至目标样本集合中,目标样本集合中的样本图像中的有效特征信息分布于第一特征提取模型重点关注的分布区域,因此第一特征提取模型可作为目标样本集合对应的目标特征提取模型,以用于重点提取目标样本集合中样本图像的有效特征信息所在的分布区域的特征信息。第二特征提取模型是基于目标样本集合训练得到的,其着重训练对目标样本集合中样
本图像的有效特征信息所在分布区域进行特征提取,模型更加收敛。对于训练样本集合筛选出目标样本集合后剩余的样本图像,这部分样本图像的有效特征信息分布在其他分布区域,在后续目标特征提取模型训练过程中,可训练出重点关注其他分布区域的目标特征提取模型,实现多种目标特征提取模型的训练,以便后续通过多种目标特征提取模型从各种类型的图像数据中提取到有效特征信息。目标特征提取模型对应的目标样本集合中的样本图像属于同一数据类型,因此可将样本图像所属的目标样本集合作为该样本图像的标签信息,通过多个目标样本集合训练出数据分类模型,以便后续根据数据分类模型确定与图像数据的特征信息类型相类似的目标样本集合,并通过目标样本集合对应的目标特征提取模型提取到图像数据中的有效特征信息,保证特征信息的可靠性,提高图像处理任务的成功率。通过多种目标特征提取模型可高效提取各种复杂场景图像的有效特征数据,提高特征提取方法的适用性。
[0119]
本技术实施例提供的分类模型训练装置可以用于执行上述实施例提供的分类模型训练方法,具备相应的功能和有益效果。同样的,本技术实施例提供的图像特征提取装置可以用于执行上述实施例提供的图像特征提取方法,具备相应的功能和有益效果。未在上述实施例中详尽描述的技术细节,可参见本技术任意实施例所提供的分类模型训练方法或图像特征提取方法。
[0120]
图11是本技术实施例提供的一种电子设备的结构示意图。该电子设备可以是分类模型训练设备或图像特征提取设备。参考图11,该电子设备包括:处理器51、存储器52、通信装置53、输入装置54及输出装置55。该电子设备中处理器31的数量可以是一个或者多个,该电子设备中的存储器52的数量可以是一个或者多个。该电子设备的处理器51、存储器52、通信装置53、输入装置54及输出装置55可以通过总线或者其他方式连接。
[0121]
存储器52作为一种计算机可读存储介质,可用于存储软件程序、计算机可执行程序以及模块,如本技术任意实施例的分类模型训练方法或图像特征提取方法对应的程序指令/模块(例如,分类模型训练装置中的第一训练模块31、第二训练模块32和第三训练模块33,或图像特征提取装置中的数据分类模块41、第一模型确定模块42和第一特征提取模块43)。存储器52可主要包括存储程序区和存储数据区,其中,存储程序区可存储操作系统、至少一个功能所需的应用程序;存储数据区可存储根据设备的使用所创建的数据等。此外,存储器52可以包括高速随机存取存储器,还可以包括非易失性存储器,例如至少一个磁盘存储器件、闪存器件、或其他非易失性固态存储器件。在一些实例中,存储器可进一步包括相对于处理器远程设置的存储器,这些远程存储器可以通过网络连接至设备。上述网络的实例包括但不限于互联网、企业内部网、局域网、移动通信网及其组合。
[0122]
通信装置53用于进行数据传输。
[0123]
处理器51通过运行存储在存储器52中的软件程序、指令以及模块,从而执行设备的各种功能应用以及数据处理,即实现上述的分类模型训练方法或图像特征提取方法。
[0124]
输入装置54可用于接收输入的数字或字符信息,以及产生与设备的用户设置以及功能控制有关的键信号输入。输出装置55可包括显示屏等显示设备。
[0125]
上述提供的电子设备可用于执行上述实施例提供的分类模型训练方法或图像特征提取方法,具备相应的功能和有益效果。
[0126]
本技术实施例还提供一种包含计算机可执行指令的存储介质,计算机可执行指令
在由计算机处理器执行时用于执行上述的分类模型训练方法或图像特征提取方法。
[0127]
存储介质——任何的各种类型的存储器设备或存储设备。术语“存储介质”旨在包括:安装介质,例如cd-rom、软盘或磁带装置;计算机系统存储器或随机存取存储器,诸如dram、ddr ram、sram、edo ram,兰巴斯(rambus)ram等;非易失性存储器,诸如闪存、磁介质(例如硬盘或光存储);寄存器或其它相似类型的存储器元件等。存储介质可以还包括其它类型的存储器或其组合。另外,存储介质可以位于程序在其中被执行的第一计算机系统中,或者可以位于不同的第二计算机系统中,第二计算机系统通过网络(诸如因特网)连接到第一计算机系统。第二计算机系统可以提供程序指令给第一计算机用于执行。术语“存储介质”可以包括驻留在不同位置中(例如在通过网络连接的不同计算机系统中)的两个或更多存储介质。存储介质可以存储可由一个或多个处理器执行的程序指令(例如具体实现为计算机程序)。
[0128]
当然,本技术实施例所提供的一种包含计算机可执行指令的存储介质,其计算机可执行指令不限于如上的分类模型训练方法或图像特征提取方法,还可以执行本技术任意实施例所提供的分类模型训练方法或图像特征提取方法中的相关操作。
[0129]
上述实施例中提供的存储介质及电子设备可执行本技术任意实施例所提供的分类模型训练方法或图像特征提取方法,未在上述实施例中详尽描述的技术细节,可参见本技术任意实施例所提供的分类模型训练方法或图像特征提取方法。
[0130]
上述仅为本技术的较佳实施例及所运用的技术原理。本技术不限于这里的特定实施例,对本领域技术人员来说能够进行的各种明显变化、重新调整及替代均不会脱离本技术的保护范围。因此,虽然通过以上实施例对本技术进行了较为详细的说明,但是本技术不仅仅限于以上实施例,在不脱离本技术构思的情况下,还可以包括更多其他等效实施例,而本技术的范围由权利要求的范围决定。
当前第1页1 2 
网友询问留言 已有0条留言
  • 还没有人留言评论。精彩留言会获得点赞!
1