[发明专利]工业品图像的自动标注方法、装置、电子设备和存储介质

说明书

本发明提供了一种工业品图像的自动标注方法、装置、电子设备和存储介质，通过获取物体的待标注图像和分类信息；对所述待标注图像进行分割处理，以分割出所述待标注物体各组成部件的独立图像；根据所述分类信息，分别对所述待标注物体各组成部件的独立图像进行识别，以得到各组成部件的类型识别结果；根据所述类型识别结果，在所述待标注图像中的各组成部件处添加对应的标签，得到初步标注图像；显示所述初步标注图像以获取修改信息，从而生成最终的标注图像；从而可对由多个部件装配而成的物体中的各部件进行自动标注，有利于提高标注效率，降低成本。

技术领域

本发明涉及图像标注技术领域，尤其涉及一种工业品图像的自动标注方法、装置、电子设备和存储介质。

背景技术

目前，工业机器人被广泛应用于工业生产中，以对物体进行抓取转移。有些物体是由多个部件装配而成的，在进行抓取时要求只能在某些部件处进行抓取，否则可能损坏物体。从而，需要对工业机器人进行训练，以使工业机器人能够识别出可抓取物件的位置。因此，有必要训练工业机器人对装配后的物体的各个部件进行识别。

为了实现对工业机器人的训练，需要对物体图片进行标注，以作为训练用的图片，标注时需要在图像中的目标位置处添加标签，标注过程一般由人工完成，人力成本和时间成本巨大。因此，需要寻求一种能够对由多个部件装配而成的物体中的各部件进行自动标注的方法和装置，以提高标注效率，降低成本。

发明内容

鉴于上述现有技术的不足之处，本申请实施例的目的在于提供一种工业品图像的自动标注方法、装置、电子设备和存储介质，能够对由多个部件装配而成的物体中的各部件进行自动标注，有利于提高标注效率，降低成本。

第一方面，本申请实施例提供一种工业品图像的自动标注方法，包括步骤：

A1.获取待标注物体的待标注图像和分类信息；所述待标注物体由多个组成部件装配而成；

A2.对所述待标注图像进行分割处理，以分割出所述待标注物体各组成部件的独立图像；

A3.根据所述分类信息，分别对所述待标注物体各组成部件的独立图像进行识别，以得到各组成部件的类型识别结果；

A4.根据所述类型识别结果，在所述待标注图像中的各组成部件处添加对应的标签，得到初步标注图像；

A5.显示所述初步标注图像以获取修改信息，从而生成最终的标注图像。

本申请实施例的工业品图像的自动标注方法，通过获取待标注物体的待标注图像和分类信息；对所述待标注图像进行分割处理，以分割出所述待标注物体各组成部件的独立图像；根据所述分类信息，分别对所述待标注物体各组成部件的独立图像进行识别，以得到各组成部件的类型识别结果；根据所述类型识别结果，在所述待标注图像中的各组成部件处添加对应的标签，得到初步标注图像；显示所述初步标注图像以获取修改信息，从而生成最终的标注图像；从而可对由多个部件装配而成的物体中的各部件进行自动标注，有利于提高标注效率，降低成本。

优选地，步骤A2包括：

提取各组成部件的轮廓线像素点位置信息；

根据所述各组成部件的轮廓线像素点位置信息，从所述待标注图像中提取对应轮廓线包围的区域图像，得到各组成部件的图像；

以预设颜色为背景颜色，生成背景图像，并把所述组成部件的图像放入所述背景图像中;

以所述组成部件的图像的外接矩形对所述背景图像进行裁剪，得到所述组成部件的独立图像。

优选地，步骤A3包括：

根据所述分类信息，从数据库中获取对应分类类别的各部件的对照图像作为对照图像集；