一种残膜回收机防缠绕挑膜装置的制 一种秧草收获机用电力驱动行走机构

目标覆盖率的获取方法、分类模型的训练方法及装置与流程

2022-09-01 06:52:55 来源:中国专利 TAG:


1.本发明涉及深度学习技术领域,特别是涉及一种目标覆盖率的获取方法、分类模型的训练方法及装置。


背景技术:

2.目前,在对基于数据种类对相关数据进行目标覆盖率统计的应用场景中,例如对商店的商品铺货率进行统计,以及对停车场的停车率进行统计。以对商店的商品铺货率进行统计为例,需要人工统计商店中所有商品的数量和商品的种类,再基于商品数量和商品种类,统计出每个商品种类对应的铺货率。
3.也就是说,在现有的目标覆盖率统计的应用场景中,存在大量的数据标注的工作,这降低了目标覆盖率统计的效率。


技术实现要素:

4.本发明实施例的目的在于提供一种目标覆盖率的获取方法、分类模型的训练方法及装置,解决目标覆盖率统计效率较低的技术问题。
5.为了解决上述技术问题,第一方面,本发明实施例提供了一种目标覆盖率的获取方法,所述方法包括:
6.获取待检测图像中第一检测区域的区域信息;
7.根据所述区域信息,确定所述第一检测区域中第一采样点分布信息;
8.根据所述第一采样点分布信息,在所述第一检测区域中截取m个检测图片;m为大于1的正整数;
9.利用目标分类模型对所述m个检测图片进行目标处理,得到所述第一检测区域对应的目标覆盖率。
10.可选地,所述根据所述区域信息,确定所述第一检测区域中第一采样点分布信息包括:
11.确定在所述第一检测区域截取的检测图片的图片规格;
12.根据所述图片规格以及所述第一检测区域的区域信息,确定所述第一检测区域中第一采样点分布信息。
13.可选地,所述区域信息包括第一检测区域的场景信息,所述确定在所述第一检测区域截取的检测图片的图片规格包括:
14.根据所述场景信息确定所述检测图片的图片规格。
15.可选地,所述区域信息包括:第一检测区域的位置信息,所述根据所述图片规格以及所述第一检测区域的区域信息,确定所述第一检测区域中第一采样点分布信息包括:
16.根据所述图片规格,确定第一采样点之间的第一间距值;
17.根据所述第一间距值以及所述第一检测区域的位置信息,确定所述第一检测区域中第一采样点分布信息。
18.可选地,所述区域信息包括第一检测区域的位置信息以及所述第一检测区域的第一面积,所述根据所述图片规格以及所述第一检测区域的区域信息,确定所述第一检测区域中第一采样点分布信息包括:
19.根据所述图片规格确定所述第一检测区域截取的检测图片对应的第二面积;
20.基于所述第一面积和所述第二面积,确定所述第一检测区域中第一采样点的采样点数量;
21.基于所述采样点数量以及所述第一检测区域的位置信息,确定所述第一检测区域中第一采样点分布信息。
22.可选地,所述基于所述采样点数量以及所述第一检测区域的位置信息,确定所述第一检测区域中第一采样点分布信息包括:
23.根据所述第一面积和所述采样点数量,确定第一采样点之间的间距值;
24.根据所述第一检测区域的坐标信息和所述间距值,确定所述第一检测区域中第一采样点分布信息。
25.可选地,所述根据所述第一采样点分布信息,在所述第一检测区域中截取m个检测图片包括:
26.以每个第一采样点为采样中心,按照预设的第一采样尺寸,对所述每个第一检测区域的图像进行截取,得到所述m个检测图片。
27.可选地,所述利用目标分类模型对所述m个检测图片进行目标处理,得到所述第一检测区域对应的目标覆盖率包括:
28.将所述m个检测图片输入至目标分类模型中进行分类,得到分类结果;所述分类结果用于表征每个检测图片对应的种类;
29.将所述m个检测图片中种类为预设种类的检测图片确定为目标检测图片;
30.基于目标检测图片的数量,得到所述第一检测区域对应的目标覆盖率。
31.可选地,所述待检测图像包括l个第一检测区域,l为小于或等于m且大于1的正整数,所述方法包括:
32.获取每个第一检测区域中检测图片的数量和目标检测图片的数量;
33.将所述每个第一检测区域中所述目标检测图片的数量与所述检测图片的数量之间的比值,确定为所述第一检测区域对应的目标覆盖率;
34.基于所述l个第一检测区域对应的目标覆盖率,确定所述待检测图像对应的目标覆盖率。
35.可选地,所述基于所述l个第一检测区域对应的目标覆盖率,确定所述待检测图像对应的目标覆盖率包括:
36.将所述l个第一检测区域对应的目标覆盖率的平均值,确定为所述待检测图像对应的目标覆盖率;或者,
37.对所述l个第一检测区域对应的目标覆盖率进行加权平均处理,得到所述待检测图像对应的目标覆盖率。
38.第二方面,本发明实施例还提供了一种分类模型的训练方法,所述方法包括:
39.获取训练数据集;所述训练数据集包括训练图片,所述训练图片为对预设的待训练图像中第二检测区域进行截取得到的图片;
40.将所述训练数据集输入至待训练的分类模型,对所述待训练的分类模型进行迭代训练,得到目标分类模型。
41.可选地,待训练图像包括至少一个第二检测区域,所述获取训练数据集之前,所述方法包括:
42.对于所述待训练图像中的每个第二检测区域,在所述第二检测区域随机生成i个第二采样点,i为大于1的正整数;
43.以每个所述第二采样点为采样中心,按照预设的第二采样尺寸,对所述第二检测区域的图像进行截取,得到i个训练图片;
44.对所述待训练图像对应的训练图片进行组合,得到所述训练数据集。
45.第三方面,本发明实施例还提供了一种目标覆盖率的获取装置,包括:
46.第一获取模块,用于获取待检测图像中第一检测区域的区域信息;
47.第一确定模块,用于根据所述区域信息,确定所述第一检测区域中第一采样点分布信息;
48.截取模块,用于根据所述第一采样点分布信息,在所述第一检测区域中截取m个检测图片;m为大于1的正整数;
49.处理模块,用于利用目标分类模型对所述m个检测图片进行目标处理,得到所述第一检测区域对应的目标覆盖率。
50.可选地,所述第一确定模块,具体用于:
51.确定在所述第一检测区域截取的检测图片的图片规格;
52.根据所述图片规格以及所述第一检测区域的区域信息,确定所述第一检测区域中第一采样点分布信息。
53.可选地,所述区域信息包括第一检测区域的场景信息,所述第一确定模块,还具体用于:
54.根据所述场景信息确定所述检测图片的图片规格。
55.可选地,所述区域信息包括:第一检测区域的位置信息,所述第一确定模块,还具体用于:
56.根据所述图片规格,确定第一采样点之间的第一间距值;
57.根据所述第一间距值以及所述第一检测区域的位置信息,确定所述第一检测区域中第一采样点分布信息。
58.可选地,所述区域信息包括第一检测区域的位置信息以及所述第一检测区域的第一面积,所述第一确定模块,还具体用于:
59.根据所述图片规格确定所述第一检测区域截取的检测图片对应的第二面积;
60.基于所述第一面积和所述第二面积,确定所述第一检测区域中第一采样点的采样点数量;
61.基于所述采样点数量以及所述第一检测区域的位置信息,确定所述第一检测区域中第一采样点分布信息。
62.可选地,所述第一确定模块,还具体用于:
63.根据所述第一面积和所述采样点数量,确定第一采样点之间的间距值;
64.根据所述第一检测区域的坐标信息和所述间距值,确定所述第一检测区域中第一
采样点分布信息。
65.可选地,所述截取模块,具体用于:
66.以每个第一采样点为采样中心,按照预设的第一采样尺寸,对所述每个第一检测区域的图像进行截取,得到所述m个检测图片。
67.可选地,所述处理模块,具体用于:
68.将所述m个检测图片输入至目标分类模型中进行分类,得到分类结果;所述分类结果用于表征每个检测图片对应的种类;
69.将所述m个检测图片中种类为预设种类的检测图片确定为目标检测图片;
70.基于目标检测图片的数量,得到所述第一检测区域对应的目标覆盖率。
71.可选地,所述待检测图像包括l个第一检测区域,l为小于或等于m且大于1的正整数,所述装置包括:
72.第二获取模块,用于获取每个第一检测区域中检测图片的数量和目标检测图片的数量;
73.第二确定模块,用于将所述每个第一检测区域中所述目标检测图片的数量与所述检测图片的数量之间的比值,确定为所述第一检测区域对应的目标覆盖率;
74.第三确定模块,用于基于所述l个第一检测区域对应的目标覆盖率,确定所述待检测图像对应的目标覆盖率。
75.可选地,所述第三确定模块,还具体用于:
76.将所述l个第一检测区域对应的目标覆盖率的平均值,确定为所述待检测图像对应的目标覆盖率;或者,
77.对所述l个第一检测区域对应的目标覆盖率进行加权平均处理,得到所述待检测图像对应的目标覆盖率。
78.第四方面,本发明实施例还提供了一种分类模型的训练装置,包括:
79.获取模块,用于获取训练数据集;所述训练数据集包括训练图片,所述训练图片为对预设的待训练图像中第二检测区域进行截取得到的图片;
80.训练模块,用于将所述训练数据集输入至待训练的分类模型,对所述待训练的分类模型进行迭代训练,得到目标分类模型。
81.可选地,待训练图像包括至少一个第二检测区域,所述装置还包括:
82.生成模块,用于对于所述待训练图像中的每个第二检测区域,在所述第二检测区域随机生成i个第二采样点,i为大于1的正整数;
83.截取模块,用于以每个所述第二采样点为采样中心,按照预设的第二采样尺寸,对所述第二检测区域的图像进行截取,得到i个训练图片;
84.组合模块,用于对所述待训练图像对应的训练图片进行组合,得到所述训练数据集。
85.第五方面,本发明实施例还提供了一种电子设备,包括处理器、通信接口、存储器和通信总线,其中,处理器,通信接口,存储器通过通信总线完成相互间的通信;
86.存储器,用于存放计算机程序;
87.处理器,用于执行存储器上所存放的程序时,实现如上所述的目标覆盖率的获取方法,或者实现如上所述的分类模型的训练方法。
88.第六方面,本发明实施例还提供了一种计算机可读存储介质,其上存储有计算机程序,该程序被处理器执行时,实现如上所述的目标覆盖率的获取方法,或者实现如上所述的分类模型的训练方法。
89.本发明实施例中,获取待检测图像中每个第一检测区域的区域信息;根据区域信息,确定第一检测区域中第一采样点分布信息;根据第一采样点分布信息,在第一检测区域中截取m个检测图片;通过上述方式对待检测图像中第一检测区域的部分图像进行截取,得到检测图片,这降低了数据准备的工作量。进一步的,利用目标分类模型对m个检测图片进行目标处理,得到第一检测区域对应的目标覆盖率。上述过程中,利用分类模型确定每个检测图片对应的种类,而不需要人工的对每个检测图片的种类进行标注,这降低了对检测图片进行数据标注的工作量,以此提高了效率。
附图说明
90.为了更清楚地说明本发明实施例或现有技术中的技术方案,下面将对实施例或现有技术描述中所需要使用的附图作简单地介绍。
91.图1为本发明实施例中目标覆盖率的获取方法的流程示意图;
92.图2为本发明实施例中目标覆盖率的获取方法的应用场景示意图之一;
93.图3为本发明实施例中目标覆盖率的获取方法的应用场景示意图之二;
94.图4为本发明实施例中分类模型的训练方法的流程示意图;
95.图5为本发明实施例中目标覆盖率的获取方法的应用场景示意图之三;
96.图6为本发明实施例中目标覆盖率的获取装置的结构示意图;
97.图7为本发明实施例中分类模型的训练装置的结构示意图;
98.图8为本发明实施例中一种电子设备的结构示意图。
具体实施方式
99.下面将结合本发明实施例中的附图,对本发明实施例中的技术方案进行描述。
100.请参阅图1,图1为本发明实施例中目标覆盖率的获取方法的流程示意图。本发明实施例提供的目标覆盖率的获取方法包括:
101.s101,获取待检测图像中第一检测区域的区域信息。
102.上述待检测图像可以是测试视频中的一帧图像,上述第一检测区域可以是待检测图像中标注的感兴趣(region of interest,roi)区域,应理解,在对商品铺货率进行统计的应用场景中,上述roi区域为商品摆放区。上述待检测图像包括至少一个roi区域,关于区域信息的具体定义请参阅后续实施例。
103.s102,根据所述区域信息,确定所述第一检测区域中第一采样点分布信息。
104.上述第一采样点分布信息包括但不限于采样点数量和采样点分布方式,具体的如何根据区域信息,确定每个第一检测区域中第一采样点分布信息的技术方案,请参阅后续实施例。
105.应理解,上述第一采样点可以随机分布于第一检测区域,也可以均匀分布于第一检测区域。为保证步骤s103中在第一检测区域中截取m个检测图片能够覆盖到所有的检测区域,在进行目标覆盖率的检测时,本发明优先的是采用均布分布方式。
106.s103,根据所述第一采样点分布信息,在所述第一检测区域中截取m个检测图片。
107.本步骤中,对每个第一检测区域进行截取,得到m个检测图片,其中,m为大于1的正整数。具体的如何对检测图片的第一检测区域进行截取处理的技术方案,请参阅后续实施例。
108.s104,利用目标分类模型对所述m个检测图片进行目标处理,得到所述第一检测区域对应的目标覆盖率。
109.本步骤中,在得到m个检测图片后,将该m个检测图片输入至训练完成的分类模型中,以对该m个检测图片进行目标处理,得到第一检测区域对应的目标覆盖率。
110.可选地,上述目标分类模型可以是深度学习分类网络模型,该目标分类模型包括但不限于mobilenet模型和resnet模型。可选地,上述目标分类模型为二分类模型。
111.本发明实施例中,获取待检测图像中第一检测区域的区域信息;根据区域信息,确定第一检测区域中第一采样点分布信息;根据第一采样点分布信息,在第一检测区域中截取m个检测图片;通过上述方式对待检测图像中第一检测区域的部分图像进行截取,得到检测图片,这降低了数据准备的工作量,同时只需对待检测图像进行标注,可以有效降低数据标注的工作量。进一步的,利用目标分类模型对m个检测图片进行目标处理,得到第一检测区域对应的目标覆盖率。
112.可选地,所述根据所述区域信息,确定所述第一检测区域中第一采样点分布信息包括:
113.确定在所述第一检测区域截取的检测图片的图片规格;
114.根据所述图片规格以及所述第一检测区域的区域信息,确定所述第一检测区域中第一采样点分布信息。
115.上述检测图片的图片规格包括但不限于检测图片的长度和宽度,上述图片规格可以是用户设定的,可以是根据目标分类模型的训练图片规格确定,还可以是根据检测图片对应的检测场景中目标检测物的规格确定的。
116.可选地,所述区域信息包括第一检测区域的场景信息,所述确定在所述第一检测区域截取的检测图片的图片规格包括:
117.根据所述场景信息确定所述检测图片的图片规格。
118.应理解,在第一检测区域的场景信息不同的情况下,检测图片的图片规格不同。具体可以理解为,不同的检测场景中包含的目标检测物类型不同,不同目标检测物的规格不同,其对应的检测图片的图片规格也不同。例如,若第一检测区域的场景信息中的检测目标为人员,则确定该检测图片的图片大小为人员的等比例缩放图;又例如,在图片规格包括图片的面积的情况下,若第一检测区域的场景信息为商品,确定检测图片的图片面积为第一面积;若第一检测区域的场景信息为车辆,确定检测图片的图片面积为第二面积;那么,第二面积大于第一面积。即不同的场景信息,对应的检测图片的图片规格不同。
119.可选地,所述区域信息包括:第一检测区域的位置信息,所述根据所述图片规格以及所述第一检测区域的区域信息,确定所述第一检测区域中第一采样点分布信息包括:
120.根据所述图片规格,确定第一采样点之间的第一间距值;
121.根据所述第一间距值以及所述第一检测区域的位置信息,确定所述第一检测区域中第一采样点分布信息。
122.本实施例中,第一采样点之间的间距值可以基于图片规格确定,如上所述,图片规格包括图片的长度和宽度。第一采样点分布的间隔可以基于检测图片的长度和宽度确定。一种可选地实施方式为第一采样点之间的第一间距值为检测图片的长度;另一种可选地实施方式为第一采样点之间的第一间距值为检测图片的宽度。例如,相邻的第一采样点的横坐标之间的距离为检测图片的长度,相邻的第一采样点的纵坐标之间的距离为检测图片的宽度。
123.本实施例中,在得到第一间距值之后,由于第一采样点是均匀分布于第一检测区域,因此可以基于第一检测区域的位置信息,将第一采样点均匀分布于第一检测区域,且设置第一采样点之间的距离为第一间距值,以此得到第一采样点分布信息。
124.可选地,所述区域信息包括第一检测区域的位置信息以及所述第一检测区域的第一面积,所述根据所述图片规格以及所述第一检测区域的区域信息,确定所述第一检测区域中第一采样点分布信息包括:
125.根据所述图片规格确定所述第一检测区域截取的检测图片对应的第二面积;
126.基于所述第一面积和所述第二面积,确定所述第一检测区域中第一采样点的采样点数量;
127.基于所述采样点数量以及所述第一检测区域的位置信息,确定所述第一检测区域中第一采样点分布信息。
128.如上所述,图片规格包括但不限于检测图片的长度和宽度,本实施例中,可以将检测图片的长度和宽度作乘法运算,得到检测图片对应的面积,将上述面积称为第二面积。
129.上述第一面积用于表征第一检测区域的面积,在得到第二面积后,一种可选地实施方式为,将第一面积与第二面积做除法运算,得到除法结果,并对该除法结果向下取整,得到第一采样点的采样点数量。
130.以下,具体阐述如何基于采样点数量以及所述第一检测区域的位置信息,确定所述第一检测区域中第一采样点分布信息。
131.本实施例中,可以基于第一检测区域的位置信息,获取第一检测区域对应的最小外界矩阵,为便于理解,请参阅图2,示出了一个第一检测区域对应的最小外界矩阵,在图2示出的应用场景中设置有一个二维直角坐标系,该坐标系的x坐标轴的朝向向右,该坐标系的y坐标轴的朝向向下。
132.将该最小外界矩阵的左上角的端点确定为第一端点,将位于最小外界矩阵的右下角的端点确定为第二端点。将第一端点的横坐标值确定为第一坐标值,该第一坐标值又称为最小横坐标值;将第一端点的纵坐标值确定为第二坐标值,该第二坐标值又称为最小纵坐标值;将第二端点的横坐标值确定为第三坐标值,该第三坐标值又称为最大横坐标值;将第二端点的纵坐标值确定为第四坐标值,该第四坐标值又称为最大纵坐标值。
133.本实施例中,可以通过以下公式计算得到第一采样点之间的间距值:
134.d=sqrt(s/k)
135.其中,d为间距值;s为第一检测区域的面积,即第一面积;k为第一采样点的采样点数量。
136.进一步的,基于上述最小外界矩阵和间距值,确定第一数列和第二数列,应理解,第一采样点的行坐标位于第一数列,且列坐标位于第二数列,以此确定第一采样点的采样
点分布信息。
137.上述第一数列可以理解为第一采样点的行坐标的取值范围,上述第二数列可以理解为第一采样点的纵坐标的取值范围。具体而言,可以通过以下伪代码计算第一数列:
138.[range(start=xmin random(0,d),end=xmax,step=d)]
[0139]
其中,符号[]表示取整操作,range()表示生成以start为起始值、end为结束值、步长step的数列,random()表示在0到d之间的随机数,xmin为第一坐标值,xmax为第三坐标值,d为间距值。
[0140]
具体而言,可以通过以下伪代码计算第二数列:
[0141]
[range(start=ymin random(0,d),end=ymax,step=d)]
[0142]
其中,符号[]表示取整操作,range()表示生成以start为起始值、end为结束值、步长step的数列,random()表示在0到d之间的随机数,ymin为第二坐标值,ymax为第四坐标值,d为间距值。
[0143]
在确定第一数列和第二数列后,将行坐标位于第一数列,纵坐标位于第二数列且位于第一检测区域内的采样点,确定为第一采样点。
[0144]
这里需要说明的是,需要说明的是,若通过该上述伪代码的计算方法生成的有效采样点数k’≠k,以实际有效采样点数k’为准。
[0145]
可选地,所述根据所述第一采样点分布信息,在所述第一检测区域中截取m个检测图片包括:
[0146]
以每个第一采样点为采样中心,按照预设的第一采样尺寸,对所述每个第一检测区域的图像进行截取,得到所述m个检测图片。
[0147]
如上所述,第一检测区域可以是待测试图像中被人工标注的roi区域,上述待测试图像包括至少一个第一检测区域。
[0148]
本实施例中,对于待测试图像中的任一第一检测区域,在第一检测区域生成均匀分布的m个第一采样点,相邻的第一采样点之间的距离为上述间距值。为便于理解,请参阅图3,在图3示出的应用场景中,第一采样点均匀分布于roi区域。
[0149]
具体的截取处理过程为:
[0150]
对于一个第一采样点,以该第一采样点为采样中心,按照预设的第一采样尺寸,对第一检测区域的图像进行截取,得到一个图片,将该图片确定为检测图片。其中,上述检测图片的中心点为第一采样点,上述第一采样尺寸具体的尺寸大小可以根据检测场景中的目标检测物的规格确定,例如检测场景中的检测物为汽车,该第一采样尺寸可以为第一尺寸;例如,检测场景中的检测物为面包,该第一采样尺寸可以为第二尺寸,其中第一尺寸大于第二尺寸。或者,上述第一采样尺寸具体的尺寸大小还可以根据目标分类模型的训练样本图片规格确定。
[0151]
本实施例中,通过对对待测试图像的第一检测区域进行截取处理,从而得到检测图片,不需要人工对待测试图像进行标注生成检测图片,以此降低了工作量,提高了目标覆盖率统计的效率。
[0152]
可选地,所述利用目标分类模型对所述m个检测图片进行目标处理,得到所述第一检测区域对应的目标覆盖率包括:
[0153]
将所述m个检测图片输入至目标分类模型中进行分类,得到分类结果;
[0154]
将所述m个检测图片中种类为预设种类的检测图片确定为目标检测图片;
[0155]
基于目标检测图片的数量,得到所述第一检测区域对应的目标覆盖率。
[0156]
本实施例中,在得到m个检测图片之后,将这m个检测图片输入至目标分类模型中进行分类,得到分类结果,其中,分类结果用于表征每个检测图片对应的种类。
[0157]
将种类为预设种类的检测图片确定为目标检测图片,并获取目标检测图片的数量。进一步的,基于目标检测图片的数量和检测图片的数量,得到第一检测区域对应的目标覆盖率。
[0158]
本实施例中,可以通过以下伪代码计算一个第一检测区域对应的目标覆盖率:
[0159]
[m1/m2*100]%
[0160]
其中,符号[]表示取整操作,m1表示该第一检测区域中的目标检测图片的数量,m2表示该第一检测区域中的检测图片的数量。
[0161]
可选地,所述待检测图像包括l个第一检测区域,l为小于或等于m且大于1的正整数,所述方法包括:
[0162]
获取每个第一检测区域中检测图片的数量和目标检测图片的数量;
[0163]
将所述每个第一检测区域中所述目标检测图片的数量与所述检测图片的数量之间的比值,确定为所述第一检测区域对应的目标覆盖率;
[0164]
基于所述l个第一检测区域对应的目标覆盖率,确定所述待检测图像对应的目标覆盖率。
[0165]
进一步的,本发明实施例可选的是,将属于同一个第一检测区域的检测图片作为一个batch(批次)输入至目标分类模型进行分类,即将同一个第一检测区域的m个检测图片一同输入到目标分类模型进行分类。进而提高使用分类模型对检测图片进行分类的效率。
[0166]
上述待检测图像包括l个第一采样区域,l为小于m且大于1的正整数。
[0167]
在得到l个第一检测区域对应的统计值之后,可以基于这l个统计值,得到待检测图像对应的目标覆盖率。
[0168]
可选地,所述基于所述l个第一检测区域对应的目标覆盖率,确定所述待检测图像对应的目标覆盖率包括:
[0169]
将所述l个第一检测区域对应的目标覆盖率的平均值,确定为所述待检测图像对应的目标覆盖率;或者,
[0170]
对所述l个第一检测区域对应的目标覆盖率进行加权平均处理,得到所述待检测图像对应的目标覆盖率。
[0171]
应理解,在对商品铺货率进行计算的场景中,上述待检测图像对应的目标覆盖率用于表征商品铺货率。
[0172]
请参阅图4,图4为本发明实施例中分类模型的训练方法的流程示意图。本发明实施例提供的分类模型的训练方法包括:
[0173]
s201,获取训练数据集。
[0174]
本实施例提供的分类模型的训练方法应用于分类模型的训练装置,上述装置可以接收用户输入的训练数据集,或者通过互联网下载训练数据集,在此并不限制获取训练数据集的具体方式。
[0175]
本步骤中,上述训练数据集包括训练图片,训练图片为对预设的待训练图像中第
二检测区域进行截取得到的图片。其中,第二检测区域为自定义设置的roi区域。
[0176]
s202,将所述训练数据集输入至待训练的分类模型,对所述待训练的分类模型进行迭代训练,得到目标分类模型。
[0177]
本步骤中的待训练的分类模型可以是深度学习分类模型,包括但不限于mobilenet模型和resnet模型,可选地,上述待训练的分类模型为二分类模型。
[0178]
可选地,在对商品铺货率进行计算的应用场景中,设置上述训练数据集包括商品数据集和背景数据集,该商品数据集中的训练图片基于对预设的包括商品的训练图像中的第二检测区域进行截取得到,该背景数据集中的训练图片基于对预设的不包括商品的训练图像中的第二检测区域进行截取得到。
[0179]
本发明可选的是,步骤s202包括:在第n次迭代训练后的分类模型对应的损失函数在预设迭代次数内相同的情况下,将所述第n次迭代训练后的分类模型确定为目标分类模型。本步骤中,n为大于1的正整数。
[0180]
本发明可选的是,步骤s202包括:在完成一个迭代训练后,对迭代训练得到的分类模型进行评估,若评估该分类模型满足输出条件,得到该目标分类模型;若评估该分类模型未满足输出条件,则继续进行迭代训练。其中该输出条件可以包括但不限于:模型精准率度、模型准确率、召回率、f1函数。
[0181]
本实施例中,通过对预设的待训练图像中的第二检测区域进行截取处理,得到训练图片,进而将训练图片作为训练分类模型的训练数据集。在上述过程中,不需要人工对训练图片进行标注,以此减少了数据标注的工作量,提高了分类模型在训练过程中的效率。
[0182]
可选地,所述待训练图像包括至少一个第二检测区域,所述获取训练数据集之前,所述方法包括:
[0183]
对于所述待训练图像中的每个第二检测区域,在所述第二检测区域生成i个第二采样点;
[0184]
以每个所述第二采样点为采样中心,按照预设的第二采样尺寸,对所述第二采样区域的图像进行截取,得到i个测试样本;
[0185]
对所述待训练图像对应的训练图片进行组合处理,得到所述训练数据集。
[0186]
如上所述,第二检测区域可以是待训练图像中被人工标注的roi区域,上述训练图片包括至少一个第二检测区域。
[0187]
本实施例中,对于训练图片中的任一第二检测区域,在第二检测区域随机生成i个第二采样点,i为大于1的正整数。为便于理解,请参阅图5,在图5示出的应用场景中,第二采样点随机分布于roi区域。
[0188]
具体的采样过程为:
[0189]
对于一个第二采样点,以该第二采样点为采样中心,按照预设的第二采样尺寸,对第二检测区域的图像进行采样,得到一个图像,将该图像确定为训练图片。其中,上述训练图片的中心点为第二采样点,上述第二采样尺寸具体的尺寸大小可以根据统计场景的不同进行自定义设置,可选地,第二采样尺寸与上述第一采样尺寸相同。
[0190]
本发明实施例还提供了一种目标覆盖率的获取装置,如图6所示,目标覆盖率的获取装置300包括:
[0191]
第一获取模块301,用于获取待检测图像中第一检测区域的区域信息;
[0192]
第一确定模块302,用于根据所述区域信息,确定所述第一检测区域中第一采样点分布信息;
[0193]
截取模块303,用于根据所述第一采样点分布信息,在所述第一检测区域中截取m个检测图片;m为大于1的正整数;
[0194]
处理模块304,用于利用目标分类模型对所述m个检测图片进行目标处理,得到所述第一检测区域对应的目标覆盖率。
[0195]
可选地,所述第一确定模块302,具体用于:
[0196]
确定在所述第一检测区域截取的检测图片的图片规格;
[0197]
根据所述图片规格以及所述第一检测区域的区域信息,确定所述第一检测区域中第一采样点分布信息。
[0198]
可选地,所述区域信息包括第一检测区域的场景信息,所述第一确定模块302,还具体用于:
[0199]
根据所述场景信息确定所述检测图片的图片规格。
[0200]
可选地,所述区域信息包括:第一检测区域的位置信息,所述第一确定模块302,还具体用于:
[0201]
根据所述图片规格,确定第一采样点之间的第一间距值;
[0202]
根据所述第一间距值以及所述第一检测区域的位置信息,确定所述第一检测区域中第一采样点分布信息。
[0203]
可选地,所述区域信息包括第一检测区域的位置信息以及所述第一检测区域的第一面积,所述第一确定模块302,还具体用于:
[0204]
根据所述图片规格确定所述第一检测区域截取的检测图片对应的第二面积;
[0205]
基于所述第一面积和所述第二面积,确定所述第一检测区域中第一采样点的采样点数量;
[0206]
基于所述采样点数量以及所述第一检测区域的位置信息,确定所述第一检测区域中第一采样点分布信息。
[0207]
可选地,所述第一确定模块302,还具体用于:
[0208]
根据所述第一面积和所述采样点数量,确定第一采样点之间的间距值;
[0209]
根据所述第一检测区域的坐标信息和所述间距值,确定所述第一检测区域中第一采样点分布信息。
[0210]
可选地,所述截取模块303,具体用于:
[0211]
以每个第一采样点为采样中心,按照预设的第一采样尺寸,对所述每个第一检测区域的图像进行截取,得到所述m个检测图片。
[0212]
可选地,所述处理模块304,具体用于:
[0213]
将所述m个检测图片输入至目标分类模型中进行分类,得到分类结果;所述分类结果用于表征每个检测图片对应的种类;
[0214]
将所述m个检测图片中种类为预设种类的检测图片确定为目标检测图片;
[0215]
基于目标检测图片的数量,得到所述第一检测区域对应的目标覆盖率。
[0216]
可选地,所述待检测图像包括l个第一检测区域,l为小于或等于m且大于1的正整数,所述装置包括:
[0217]
获取模块,用于获取每个第一检测区域中检测图片的数量和目标检测图片的数量;
[0218]
第二确定模块,用于将所述每个第一检测区域中所述目标检测图片的数量与所述检测图片的数量之间的比值,确定为所述第一检测区域对应的目标覆盖率;
[0219]
第三确定模块,用于基于所述l个第一检测区域对应的目标覆盖率,确定所述待检测图像对应的目标覆盖率。
[0220]
可选地,所述第三确定模块,还具体用于:
[0221]
将所述l个第一检测区域对应的目标覆盖率的平均值,确定为所述待检测图像对应的目标覆盖率;或者,
[0222]
对所述l个第一检测区域对应的目标覆盖率进行加权平均处理,得到所述待检测图像对应的目标覆盖率。
[0223]
本发明实施例还提供了一种分类模型的训练装置,如图7所示,分类模型的训练装置400包括:
[0224]
获取模块401,用于获取训练数据集;
[0225]
训练模块402,用于将所述训练数据集输入至待训练的分类模型,对所述待训练的分类模型进行迭代训练,得到目标分类模型。
[0226]
可选地,所述分类模型的训练装置400还包括:
[0227]
生成模块,用于对于所述待训练图像中的每个第二检测区域,在所述第二检测区域随机生成i个第二采样点,i为大于1的正整数;
[0228]
截取模块,用于以每个所述第二采样点为采样中心,按照预设的第二采样尺寸,对所述第二检测区域的图像进行截取,得到i个训练图片;
[0229]
组合模块,用于对所述待训练图像对应的训练图片进行组合,得到所述训练数据集。
[0230]
本发明实施例还提供了一种电子设备,如图8所示,包括处理器501、通信接口502、存储器503和通信总线504,其中,处理器501,通信接口502,存储器503通过通信总线504完成相互间的通信。
[0231]
存储器503,用于存放计算机程序;
[0232]
处理器501,用于执行存储器503上所存放的程序时,所述计算机程序被所述处理器501执行时,用于执行上述实施例中任一实施例所述的目标覆盖率的获取方法,或者执行上述实施例中任一实施例所述的分类模型的训练方法。
[0233]
上述电子设备提到的通信总线可以是外设部件互连标准(peripheral component interconnect,简称pci)总线或扩展工业标准结构(extended industry standard architecture,简称eisa)总线等。该通信总线可以分为地址总线、数据总线、控制总线等。为便于表示,图中仅用一条粗线表示,但并不表示仅有一根总线或一种类型的总线。
[0234]
通信接口用于上述终端与其他设备之间的通信。
[0235]
存储器可以包括随机存取存储器(random access memory,简称ram),也可以包括非易失性存储器(non-volatile memory),例如至少一个磁盘存储器。可选的,存储器还可以是至少一个位于远离前述处理器的存储装置。
[0236]
上述的处理器可以是通用处理器,包括中央处理器(central processing unit,
简称cpu)、网络处理器(network processor,简称np)等;还可以是数字信号处理器(digital signal processing,简称dsp)、专用集成电路(application specific integrated circuit,简称asic)、现场可编程门阵列(field-programmable gate array,简称fpga)或者其他可编程逻辑器件、分立门或者晶体管逻辑器件、分立硬件组件。
[0237]
在本发明提供的又一实施例中,还提供了一种计算机可读存储介质,该计算机可读存储介质中存储有指令,当其在计算机上运行时,使得计算机执行上述实施例中任一实施例所述的目标覆盖率的获取方法,或者执行上述实施例中任一实施例所述的分类模型的训练方法。
[0238]
在本发明提供的又一实施例中,还提供了一种包含指令的计算机程序产品,当其在计算机上运行时,使得计算机执行上述实施例中任一实施例所述的目标覆盖率的获取方法,或者执行上述实施例中任一实施例所述的分类模型的训练方法。
[0239]
在上述实施例中,可以全部或部分地通过软件、硬件、固件或者其任意组合来实现。当使用软件实现时,可以全部或部分地以计算机程序产品的形式实现。所述计算机程序产品包括一个或多个计算机指令。在计算机上加载和执行所述计算机程序指令时,全部或部分地产生按照本发明实施例所述的流程或功能。所述计算机可以是通用计算机、专用计算机、计算机网络、或者其他可编程装置。所述计算机指令可以存储在计算机可读存储介质中,或者从一个计算机可读存储介质向另一个计算机可读存储介质传输,例如,所述计算机指令可以从一个网站站点、计算机、服务器或数据中心通过有线(例如同轴电缆、光纤、数字用户线(dsl))或无线(例如红外、无线、微波等)方式向另一个网站站点、计算机、服务器或数据中心进行传输。所述计算机可读存储介质可以是计算机能够存取的任何可用介质或者是包含一个或多个可用介质集成的服务器、数据中心等数据存储设备。所述可用介质可以是磁性介质,(例如,软盘、硬盘、磁带)、光介质(例如,dvd)、或者半导体介质(例如固态硬盘solid state disk(ssd))等。
[0240]
需要说明的是,在本文中,诸如第一和第二等之类的关系术语仅仅用来将一个实体或者操作与另一个实体或操作区分开来,而不一定要求或者暗示这些实体或操作之间存在任何这种实际的关系或者顺序。而且,术语“包括”、“包含”或者其任何其他变体意在涵盖非排他性的包含,从而使得包括一系列要素的过程、方法、物品或者设备不仅包括那些要素,而且还包括没有明确列出的其他要素,或者是还包括为这种过程、方法、物品或者设备所固有的要素。在没有更多限制的情况下,由语句“包括一个
……”
限定的要素,并不排除在包括所述要素的过程、方法、物品或者设备中还存在另外的相同要素。
[0241]
本说明书中的各个实施例均采用相关的方式描述,各个实施例之间相同相似的部分互相参见即可,每个实施例重点说明的都是与其他实施例的不同之处。尤其,对于系统实施例而言,由于其基本相似于方法实施例,所以描述的比较简单,相关之处参见方法实施例的部分说明即可。
[0242]
以上所述仅为本发明的较佳实施例而已,并非用于限定本发明的保护范围。凡在本发明的精神和原则之内所作的任何修改、等同替换、改进等,均包含在本发明的保护范围内。
再多了解一些

本文用于企业家、创业者技术爱好者查询,结果仅供参考。

发表评论 共有条评论
用户名: 密码:
验证码: 匿名发表

相关文献