(19)国家知识产权局 (12)发明 专利申请 (10)申请公布号 (43)申请公布日 (21)申请 号 202210582437.5 (22)申请日 2022.05.26 (71)申请人 苏州佳智彩光电科技有限公司 地址 215000 江苏省苏州市吴江区江陵街 道交通路1 188号 (72)发明人 扈健玮　 (74)专利代理 机构 苏州汇诚汇智专利代理事务 所(普通合伙) 32623 专利代理师 张聪 (51)Int.Cl. G06V 10/00(2022.01) G06T 7/00(2017.01) G06T 7/73(2017.01) G06N 20/00(2019.01) (54)发明名称 一种工业图像目标检测图像训练集的采样 方法和存 储介质 (57)摘要 本发明公开了一种工业图像目标检测图像 训练集的采集方法和存储介质， 方法包括每张待 训练图像的目标采集， 对待训练图像并进行标 注， 将待训练图像进行标注形成n个矩形图像块， 形成坐标集Rect； 获取截取子图在待训练图像上 的坐标； 排除坐标集Rect中不在截取子图坐标范 围内的矩形图像块的坐标， 并修正坐标集Rect中 剩余矩形图像块的坐标值， 得到新的坐标集 Rect’； 获取采样图像， 以所述截取子图的坐标在 所述待训练图像上截取形成采样图像， 其中采样 图像中的矩形图像块的坐标按坐标集Rect ’修 正。 既能仅对原始图像集做一次标注， 又能不断 产生分割位置变化， 而且能够通过参数自动控制 纯背景子图和带目标子图类别比例。 权利要求书2页 说明书5页 附图3页 CN 114972702 A 2022.08.30 CN 114972702 A 1.一种工业图像目标检测图像训练集的采集方法， 其特征在于： 用于对训练集中的每 张待训练图像进行目标采集， 每张待训练图像的目标采集包括如下步骤， S1、 对待训练图像并进行标注， 将待训练图像进行标注形成n个矩形图像块， 记录每个 所述矩形图像块的坐标， 每个所述矩形图像块的坐标形成坐标集Rect， 其中n为大于等于0 的整数； S2、 获取截取子 图在待训练图像上的坐标， 在待训练图像上随机采样形成矩形截取子 图， 所述截取子图为未含有目标对 象的背景子图或含有目标对 象的对象子图， 所述背景子 图和对象子图的截取比例可调节， 所述截取子图在待训练图像上的坐标在待训练图像范围 内； S3、 排除坐标集Rect中不在截取子图坐标范围内的矩形图像块的坐标， 并修正坐标集 Rect中剩余矩形图像块的坐标值， 得到新的坐标集Rect ’； S4、 获取采样图像， 以所述截取子图的坐标在所述待训练图像上截取形成采样图像， 其 中采样图像中的矩形图像块的坐标按坐标集Rect ’修正。 2.根据权利要求1所述的工业图像目标检测图像训练集的采集方法， 其特征在于： 所述 矩形图像块的坐标为(x1,y1,x2,y2)， 其中(x1,y1)、 (x2,y2)为矩形图像块斜对角的第一顶 点和第二顶点的坐标值， 所述坐标集Rect＝{(x1,y1,x2,y2)1,(x1,y1,x2,y2)2,……(x1, y1,x2,y2)n}。 3.根据权利要求2所述的工业图像目标检测图像训练集的采集方法， 其特征在于： 所述 S2具体包括， S21、 随机产生截取子图的第一顶点在待训练图像上的坐标(x3,y3)； S22、 计算截取子图第二顶点在待训练图像上坐标(x4,y4)， 其中第一顶点和第二顶点 为矩形对角线上的两个顶点， x4＝x3+ws,y4＝y3+hs， 其中ws为截取子图的宽度， hs为截取 子图的高度； S23、 判断截取子图在待训练图像上的坐标是否在待训练图像内， 若否， 执行步骤S24, 若是， 执行步骤S25； S24、 通过修正算法对S22中的(x3,y3)和(x4,y4)进行修正， 保证修整后的截取子图坐 标在待训练图像范围内， 并将修正后的(x3,y3,x4,y4)作为截取子图在待训练图像上的坐 标； S25、 将S2 2中的(x3,y3， x4,y4)作为截取子图在待训练图像上的坐标。 4.根据权利要求3所述的工业图像目标检测图像训练集的采集方法， 其特征在于： 所述 S21具体包括 生成在[0， 1]上均匀分布的随机数a并与预置的阈值b比较， 若 随机数a>阈值b， 则(x3, y3)的随机范围为x3∈[(x1+x2)i/2‑ws， (x1+x2)i/2],y3∈[(y1+y2)i/2‑hs， (y1+y2)i/2]， 其中(x1,y1,x2,y2)i为在坐标集Rect中随机选取一个矩形图像块的坐标； 若随机数a≤阈 值b， 则(x3,y3)的随机范围为x3∈[ ‑ws/2， w‑ws/2],y3∈[ ‑hs/2， h‑hs/2],其中w为待训练 图像的宽度， h为待训练图像的高度。 5.根据权利要求4所述的工业图像目标检测图像训练集的采集方法， 其特征在于： 所述 阈值b为超参数， 表示期望截 取子图内包含目标的概率， 所述范围b∈[0,1]， 所述阈值b值越 小， 所述截取子图内为背景子图的概率越大； 所述阈值b值越大， 所述截取子图内包含目标权　利　要　求　书 1/2 页 2 CN 114972702 A 2的概率越大。 6.根据权利要求3所述的工业图像目标检测图像训练集的采集方法， 其特征在于： 所述 S3具体包括 遍历坐标集Rect中的元素， 对一个矩形图像块的坐标值(x1,y1,x2,y2)i,若其中(x1)i> x4,或(y1)i>y4， 或(x2)i>x3， 或(y2)i<y3,则从坐标集Rect中去除此矩形图像块的坐标(x1, y1,x2,y2)i； 否则修正此矩形图像块坐标值， 令(x1) ’i＝max(0,x1 ‑x3),(y1) ’i＝max(0,y1 ‑y3), (x2)’i＝min(w,x2 ‑x3),(y2) ’i＝min(h,y2 ‑y3)， 坐标值(x1 ’,y1’,x2’,y2’)i即为矩形图像 块修正后的坐标值， 形成坐标集Rect ’。 7.根据权利要求4所述的工业图像目标检测图像训练集的采集方法， 其特征在于： 所述 S24中的修 正算法具体包括 若x3<0， 则修正x3＝0,x4＝ws； 若y3<0， 则修正y3＝0,y4＝hs； 若x4>w， 则修正x3＝w ‑ ws,x4＝w； 若y4<0， 则修正y3＝h‑hs,y4＝h。 8.一种存储有程序的计算机可读取存储介质， 其特征在于： 所述计算机的程序在运行 时执行如权利要求1 ‑7任一所述的工业图像目标检测图像训练集的采集方法。权　利　要　求　书 2/2 页 3 CN 114972702 A 3