[发明专利]布局模型训练方法、装置以及设备

说明书

本公开提供了一种布局模型训练方法、装置以及设备，涉及人工智能技术领域，具体涉及工业、数据处理、深度学习等技术领域。该方法的一具体实施方式包括：获取样本部件集合和样本布局图，其中，样本部件集合包括需要布局到样本布局图中的样本部件，且可拼装成样本产品；提取样本部件集合的样本特征张量和样本布局图的样本布局矩阵；利用样本部件集合的样本特征张量和样本布局图的样本布局矩阵对原始模型进行训练，得到布局模型，其中，布局模型用于从部件集合中选取下一个布局到布局图中的部件。该实施方式以部件特征和布局图状态感知为输入，训练出能够从部件集合中选取下一个布局到布局图的部件的布局模型，使得布局模型具有出色的布局能力。

技术领域

本公开涉及人工智能技术领域，具体涉及工业、数据处理、深度学习技术领域。

背景技术

板材切割是现代工业生产的经典环节。通过对木板、钢板等原材料板材进行开料、排钻、封边等处理，可得到待拼装的成品产品，例如橱柜、衣柜、浴室柜、木门等等。在这些工艺当中，开料是所有工艺的基础。将板材通过电子锯、加工中心、激光切割台、甚至人工的手推锯等设备切割成一个一个待加工部件。其中，通过在板材上排布待开料的部件，使得在切割效率足够高的情况下，获取更高的原材料板材的利用率，这是板材切割所要完成的根本目标。

发明内容

本公开实施例提出了一种布局模型训练方法、装置、设备、存储介质以及程序产品。

第一方面，本公开实施例提出了一种布局模型训练方法，包括：获取样本部件集合和样本布局图，其中，样本部件集合包括需要布局到样本布局图中的样本部件，且可拼装成样本产品；提取样本部件集合的样本特征张量和样本布局图的样本布局矩阵；利用样本部件集合的样本特征张量和样本布局图的样本布局矩阵对原始模型进行训练，得到布局模型，其中，布局模型用于从部件集合中选取下一个布局到布局图中的部件。

第二方面，本公开实施例提出了一种部件布局方法，包括：获取部件集合和布局图，其中，部件集合包括需要布局到布局图中的部件，可拼装成产品；提取部件集合的特征张量和布局图的布局矩阵；将部件集合的特征张量和布局图的布局矩阵输入到布局模型，得到目标部件，其中，布局模型是采用第一方面所述的方法训练得到的，目标部件是下一个布局到布局图中的部件；基于目标部件的布局信息，将目标部件布局到布局图中，以及将目标部件从部件集合中删除。

第三方面，本公开实施例提出了一种布局模型训练装置，包括：获取模块，被配置成获取样本部件集合和样本布局图，其中，样本部件集合包括需要布局到样本布局图中的样本部件，且可拼装成样本产品；提取模块，被配置成提取样本部件集合的样本特征张量和样本布局图的样本布局矩阵；训练模块，被配置成利用样本部件集合的样本特征张量和样本布局图的样本布局矩阵对原始模型进行训练，得到布局模型，其中，布局模型用于从部件集合中选取下一个布局到布局图中的部件。

第四方面，本公开实施例提出了一种部件布局装置，包括：获取模块，被配置成获取部件集合和布局图，其中，部件集合包括需要布局到布局图中的部件，可拼装成产品；提取模块，被配置成提取部件集合的特征张量和布局图的布局矩阵；输入模块，被配置成将部件集合的特征张量和布局图的布局矩阵输入到布局模型，得到目标部件，其中，布局模型是采用第三方面所述的装置训练得到的，目标部件是下一个布局到布局图中的部件；布局模块，被配置成基于目标部件的布局信息，将目标部件布局到布局图中，以及将目标部件从部件集合中删除。

第五方面，本公开实施例提出了一种电子设备，包括：至少一个处理器；以及与至少一个处理器通信连接的存储器；其中，存储器存储有可被至少一个处理器执行的指令，指令被至少一个处理器执行，以使至少一个处理器能够执行如第一方面中任一实现方式描述的方法或第二方面中任一实现方式描述的方法。

第六方面，本公开实施例提出了一种存储有计算机指令的非瞬时计算机可读存储介质，计算机指令用于使计算机执行如第一方面中任一实现方式描述的方法或第二方面中任一实现方式描述的方法。