本发明专利技术提供了一种工业品图像的自动标注方法、装置、电子设备和存储介质，通过获取物体的待标注图像和分类信息；对所述待标注图像进行分割处理，以分割出所述待标注物体各组成部件的独立图像；根据所述分类信息，分别对所述待标注物体各组成部件的独立图像进行识别，以得到各组成部件的类型识别结果；根据所述类型识别结果，在所述待标注图像中的各组成部件处添加对应的标签，得到初步标注图像；显示所述初步标注图像以获取修改信息，从而生成最终的标注图像；从而可对由多个部件装配而成的物体中的各部件进行自动标注，有利于提高标注效率，降低成本。降低成本。降低成本。
【技术实现步骤摘要】
工业品图像的自动标注方法、装置、电子设备和存储介质
[0001]本专利技术涉及图像标注
，尤其涉及一种工业品图像的自动标注方法、装置、电子设备和存储介质。
技术介绍
[0002]目前，工业机器人被广泛应用于工业生产中，以对物体进行抓取转移。有些物体是由多个部件装配而成的，在进行抓取时要求只能在某些部件处进行抓取，否则可能损坏物体。从而，需要对工业机器人进行训练，以使工业机器人能够识别出可抓取物件的位置。因此，有必要训练工业机器人对装配后的物体的各个部件进行识别。
[0003]为了实现对工业机器人的训练，需要对物体图片进行标注，以作为训练用的图片，标注时需要在图像中的目标位置处添加标签，标注过程一般由人工完成，人力成本和时间成本巨大。因此，需要寻求一种能够对由多个部件装配而成的物体中的各部件进行自动标注的方法和装置，以提高标注效率，降低成本。
技术实现思路
[0004]鉴于上述现有技术的不足之处，本申请实施例的目的在于提供一种工业品图像的自动标注方法、装置、电子设备和存储介质，能够对由多个部件装配而成的物体中的各部件进行自动标注，有利于提高标注效率，降低成本。
[0005]第一方面，本申请实施例提供一种工业品图像的自动标注方法，包括步骤：A1.获取物体的待标注图像和分类信息；所述物体由多个组成部件装配而成；A2.对所述待标注图像进行分割处理，以分割出所述待标注物体各组成部件的独立图像；A3.根据所述分类信息，分别对所述待标注物体各组成部件的独立图像进行识别，以得到各组成部件的类型识别结果；A4.根据所述类型识别结果，在所述待标注图像中的各组成部件处添加对应的标签，得到初步标注图像；A5.显示所述初步标注图像以获取修改信息，从而生成最终的标注图像。
[0006]本申请实施例的工业品图像的自动标注方法，通过获取物体的待标注图像和分类信息；对所述待标注图像进行分割处理，以分割出所述待标注物体各组成部件的独立图像；根据所述分类信息，分别对所述待标注物体各组成部件的独立图像进行识别，以得到各组成部件的类型识别结果；根据所述类型识别结果，在所述待标注图像中的各组成部件处添加对应的标签，得到初步标注图像；显示所述初步标注图像以获取修改信息，从而生成最终的标注图像；从而可对由多个部件装配而成的物体中的各部件进行自动标注，有利于提高标注效率，降低成本。
[0007]优选地，步骤A2包括：提取各组成部件的轮廓线像素点位置信息；
根据所述各组成部件的轮廓线像素点位置信息，从所述待标注图像中提取对应轮廓线包围的区域图像，得到各组成部件的图像；以预设颜色为背景颜色，生成背景图像，并把所述组成部件的图像放入所述背景图像中;以所述组成部件的图像的外接矩形对所述背景图像进行裁剪，得到所述组成部件的独立图像。
[0008]优选地，步骤A3包括：根据所述分类信息，从数据库中获取对应分类类别的各部件的对照图像作为对照图像集；用所述对照图像集中的各部件的对照图像与待识别的组成部件的独立图像进行匹配，得到对照图像集中的各部件的对照图像与待识别的组成部件的独立图像之间的匹配度数据；以匹配度数据最大的对照图像对应的部件类型作为相应组成部件的类型识别结果。
[0009]优选地，步骤A4包括：根据各组成部件的轮廓线像素点位置信息获取各组成部件在所述待标注图像中的重心点位置信息；在所述重心点位置处添加对应的标签。
[0010]优选地，步骤A5包括：在接收到对显示的初步标注图像中的组成部件的第一操作信息时，在显示屏上显示所述组成部件的匹配度数据前N大的对照图像的部件类型列表；其中，N是预设的正整数值，且N大于1；根据用户对所述部件类型列表的选择操作，用被选中的部件类型作为对应组成部件的修正后部件类型，并根据所述修正后部件类型更改对应的标签。
[0011]优选地，步骤A5包括：在接收到对显示的初步标注图像中的组成部件的第二操作信息时，在显示屏上显示文本框；根据用户在所述文本框中输入的部件类型，更改对应的标签。
[0012]优选地，所述根据用户在所述文本框中输入的部件类型，更改对应的标签的步骤包括：判断输入的部件类型是否为数据库中已有的部件类型；若是，则根据所述部件类型查找对应的标签，并用该标签替换所述初步标注图像中对应组成部件的标签；若否，则把对应组成部件的独立图像存储到所述数据库中作为所述部件类型的对照图像，并生成对应的标签存储到所述数据库中，并用该标签替换所述初步标注图像中对应组成部件的标签。
[0013]第二方面，本申请实施例提供一种工业品图像的自动标注装置，包括：第一获取模块，用于获取物体的待标注图像和分类信息；所述物体由多个组成部件装配而成；
分割模块，用于对所述待标注图像进行分割处理，以分割出所述待标注物体各组成部件的独立图像；识别模块，用于根据所述分类信息，分别对所述待标注物体各组成部件的独立图像进行识别，以得到各组成部件的类型识别结果；标注模块，用于根据所述类型识别结果，在所述待标注图像中的各组成部件处添加对应的标签，得到初步标注图像；修改模块，用于显示所述初步标注图像以获取修改信息，从而生成最终的标注图像。
[0014]本申请实施例的工业品图像的自动标注装置，通过获取物体的待标注图像和分类信息；对所述待标注图像进行分割处理，以分割出所述待标注物体各组成部件的独立图像；根据所述分类信息，分别对所述待标注物体各组成部件的独立图像进行识别，以得到各组成部件的类型识别结果；根据所述类型识别结果，在所述待标注图像中的各组成部件处添加对应的标签，得到初步标注图像；显示所述初步标注图像以获取修改信息，从而生成最终的标注图像；从而可对由多个部件装配而成的物体中的各部件进行自动标注，有利于提高标注效率，降低成本。
[0015]第三方面，本申请实施例提供一种电子设备，包括处理器和存储器，所述存储器中存储有计算机程序，所述处理器通过调用所述存储器中存储的所述计算机程序，用于执行如所述的工业品图像的自动标注方法的步骤。
[0016]第四方面，本申请实施例提供一种存储介质，其上存储有计算机程序，所述计算机程序被处理器执行时运行如所述的工业品图像的自动标注方法的步骤。
[0017]有益效果：本申请实施例提供的工业品图像的自动标注方法、装置、电子设备和存储介质，通过获取物体的待标注图像和分类信息；对所述待标注图像进行分割处理，以分割出所述待标注物体各组成部件的独立图像；根据所述分类信息，分别对所述待标注物体各组成部件的独立图像进行识别，以得到各组成部件的类型识别结果；根据所述类型识别结果，在所述待标注图像中的各组成部件处添加对应的标签，得到初步标注图像；显示所述初步标注图像以获取修改信息，从而生成最终的标注图像；从而可对由多个部件装配而成的物体中的各部件进行自动标注，有利于提高标注效率，降低成本。
附图说明
[0018]图1为本申请实施例提供的工业品图像的自动标注方法的流程图。
[0019]图2为本申请实施例提供的工业品图像的自动标注装置的结构示意图本文档来自技高网...
【技术保护点】
【技术特征摘要】
1.一种工业品图像的自动标注方法，其特征在于，包括步骤：A1.获取物体的待标注图像和分类信息；所述物体由多个组成部件装配而成；A2.对所述待标注图像进行分割处理，以分割出所述待标注物体各组成部件的独立图像；A3.根据所述分类信息，分别对所述待标注物体各组成部件的独立图像进行识别，以得到各组成部件的类型识别结果；A4.根据所述类型识别结果，在所述待标注图像中的各组成部件处添加对应的标签，得到初步标注图像；A5.显示所述初步标注图像以获取修改信息，从而生成最终的标注图像。2.根据权利要求1所述的工业品图像的自动标注方法，其特征在于，步骤A2包括：提取各组成部件的轮廓线像素点位置信息；根据所述各组成部件的轮廓线像素点位置信息，从所述待标注图像中提取对应轮廓线包围的区域图像，得到各组成部件的图像；以预设颜色为背景颜色，生成背景图像，并把所述组成部件的图像放入所述背景图像中;以所述组成部件的图像的外接矩形对所述背景图像进行裁剪，得到所述组成部件的独立图像。3.根据权利要求1所述的工业品图像的自动标注方法，其特征在于，步骤A3包括：根据所述分类信息，从数据库中获取对应分类类别的各部件的对照图像作为对照图像集；用所述对照图像集中的各部件的对照图像与待识别的组成部件的独立图像进行匹配，得到对照图像集中的各部件的对照图像与待识别的组成部件的独立图像之间的匹配度数据；以匹配度数据最大的对照图像对应的部件类型作为相应组成部件的类型识别结果。4.根据权利要求2所述的工业品图像的自动标注方法，其特征在于，步骤A4包括：根据各组成部件的轮廓线像素点位置信息获取各组成部件在所述待标注图像中的重心点位置信息；在所述重心点位置处添加对应的标签。5.根据权利要求3所述的工业品图像的自动标注方法，其特征在于，步骤A5包括：在接收到对显示的初步标注图像中的组成部件的第一操作信息时，在显示屏上显示所述组成部件的匹配度数据前N大的对照图像的部件类型列表；其中，N是预设的正整数值，且N大于1；根据用户对所述部件类...
【专利技术属性】
技术研发人员：张校志，王展，林才纺，杨远达，
申请(专利权)人：季华实验室，
类型：发明
国别省市：
还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。