【技术实现步骤摘要】
本公开涉及图像处理，尤其涉及一种图像标注方法、模型训练方法、装置、设备及介质。
技术介绍
1、现有训练图像分割模型的过程中，需要使用大量带有标注信息的图像作为训练数据，而这些作为训练数据的图像，都需要预先由人工进行标注，如选取目标轮廓或目标标注框等，由于需要标注的图像数量非常大，因此人工标注的时间也非常长。虽然有人提出了，仅以少量标注数据对图像分割模型进行预训练，之后再利用该模型对剩余的图像进行自动分割标注，进而得到全量训练数据的方法；但仅以少量标注数据进行训练得到图像分割模型，其输出准确性较低，进而导致后续通过自动分割标注得到的训练数据的准确性也不高，利用这样的全量训练数据训练出来的模型，其输出准确性依然无法得到保障。
2、因此如何在人工标注的过程中，降低人工标注中的工作量，减少人工标注的时间就成为亟待解决的问题。
技术实现思路
1、为了解决上述技术问题，本公开提供了一种图像标注方法、模型训练方法、装置、设备及介质，以提高图像标注的效率。
2、第一方面，本公开实施例提供一种图像标注方法，包括：
3、展示原始图像，所述原始图像中包括待标注的目标主体；
4、响应于用户对于所述目标主体的点击操作，确定所述点击操作在原始图像中对应的至少一个目标位置；
5、将带有所述目标位置信息的原始图像输入预先训练好的图像分割模型中；
6、获取所述图像分割模型输出的对于所述目标主体的图像分割结果；
7、对所述图像分割结果
8、在一些实施例中，所述展示原始图像之后，所述方法还包括：
9、响应于用户对于所述原始图像中目标主体的标签输入操作，获取所述目标主体对应的标签信息。
10、在一些实施例中，所述对所述图像分割结果添加所述目标主体对应的标签信息，得到所述原始图像的标注结果，包括：
11、对所述图像分割结果添加所述目标主体对应的标签信息，得到所述原始图像的语义掩膜图；
12、对所述语义掩膜图进行格式转换，得到所述原始图像的标注结果。
13、在一些实施例中，所述展示原始图像之后，所述方法还包括：
14、根据所述原始图像对应的图像处理任务，确定是否需要进行自动图像标注；
15、若是，则将所述原始图像输入预先训练好的图像检测模型中；
16、获取所述图像检测模型输出的所述原始图像的标注结果。
17、第二方面，本公开实施例提供一种模型训练方法，包括：
18、获取训练图像以及所述训练图像中的目标区域；
19、根据所述目标区域，生成所述训练图像的标注信息，所述标注信息至少包括目标区域内至少一个第一标注点的点信息，以及所述目标区域外至少一个第二标注点的点信息；
20、将所述训练图像以及所述训练图像的标注信息作为图像分割模型的输入，将所述目标区域作为所述图像分割模型的输出，对所述图像分割模型进行训练。
21、在一些实施例中，所述根据所述目标区域，生成所述训练图像的标注信息，包括：
22、在所述训练图像中生成预设数量的预设标注点，所述预设标注点包括所述目标区域内的至少一个第一标注点以及所述目标区域外的至少一个第二标注点；
23、针对所述预设标注点中的目标标注点，计算所述目标标注点与其他预设标注点之间的相对距离，得到所述目标标注点对应的相对距离向量；
24、将所述相对距离向量作为所述目标标注点对应的点信息。
25、第三方面，本公开实施例提供一种图像标注装置，包括：
26、展示模块，用于展示原始图像，所述原始图像中包括待标注的目标主体；
27、第一确定模块，用于响应于用户对于所述目标主体的点击操作，确定所述点击操作在原始图像中对应的至少一个目标位置；
28、输入模块，用于将带有所述目标位置信息的原始图像输入预先训练好的图像分割模型中；
29、第一获取模块，用于获取所述图像分割模型输出的对于所述目标主体的图像分割结果；
30、第二获取模块，用于对所述图像分割结果添加所述目标主体对应的标签信息，得到所述原始图像的标注结果。
31、第四方面，本公开实施例提供一种模型训练装置，包括：
32、第四获取模块，用于获取训练图像以及所述训练图像中的目标区域；
33、生成模块，用于根据所述目标区域，生成所述训练图像的标注信息，所述标注信息至少包括目标区域内至少一个第一标注点的点信息，以及所述目标区域外至少一个第二标注点的点信息；
34、训练模块，用于将所述训练图像以及所述训练图像的标注信息作为图像分割模型的输入，将所述目标区域作为所述图像分割模型的输出，对所述图像分割模型进行训练。
35、第五方面，本公开实施例提供一种电子设备，包括：
36、存储器；
37、处理器；以及
38、计算机程序；
39、其中，所述计算机程序存储在所述存储器中，并被配置为由所述处理器执行以实现如第一方面或第二方面所述的方法。
40、第六方面，本公开实施例提供一种计算机可读存储介质，其上存储有计算机程序，所述计算机程序被处理器执行以实现第一方面或第二方面所述的方法。
41、本公开实施例提供的图像标注方法、模型训练方法、装置、设备及介质，通过交互式的标注方法，用户仅需对待标注的目标主体进行点击操作，即可完成对目标主体的分割标注，有效提高了图像标注方法的效率。
本文档来自技高网...【技术保护点】
1.一种图像标注方法，其特征在于，所述方法包括：
2.根据权利要求1所述的方法，其特征在于，所述展示原始图像之后，所述方法还包括：
3.根据权利要求1所述的方法，其特征在于，所述对所述图像分割结果添加所述目标主体对应的标签信息，得到所述原始图像的标注结果，包括：
4.根据权利要求1所述的方法，其特征在于，所述展示原始图像之后，所述方法还包括：
5.一种模型训练方法，其特征在于，所述模型应用于如权利要求1-4中任一项所述的图像标注方法中，所述模型训练方法包括：
6.根据权利要求5所述的方法，其特征在于，所述根据所述目标区域，生成所述训练图像的标注信息，包括：
7.一种图像标注装置，其特征在于，所述装置包括：
8.一种模型训练装置，其特征在于，所述装置包括：
9.一种电子设备，其特征在于，包括：
10.一种计算机可读存储介质，其上存储有计算机程序，其特征在于，所述计算机程序被处理器执行时实现如权利要求1-4或权利要求5-6中任一项所述的方法。
【技术特征摘要】
1.一种图像标注方法，其特征在于，所述方法包括：
2.根据权利要求1所述的方法，其特征在于，所述展示原始图像之后，所述方法还包括：
3.根据权利要求1所述的方法，其特征在于，所述对所述图像分割结果添加所述目标主体对应的标签信息，得到所述原始图像的标注结果，包括：
4.根据权利要求1所述的方法，其特征在于，所述展示原始图像之后，所述方法还包括：
5.一种模型训练方法，其特征在于，所述模型应用于如权利要求1-4中任一项所述的图像标注...
【专利技术属性】
技术研发人员：邵慧翔，张赵行，刘永春，连理国，
申请(专利权)人：北京罗克维尔斯科技有限公司，
类型：发明
国别省市：
还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。