本申请涉及图像处理领域,特别涉及一种图像标注方法、装置、设备及介质。
背景技术:
1、图像标注是指对图像中的目标实体进行标记、注释和描述的过程。通常情况下,图像标注的目的是从图像中分割出目标实体,再为目标实体添加标签。图像标注在计算机视觉、机器学习、自然语言处理等领域都有广泛的应用。
2、相关技术中,提供有图像标注工具(如labelimage、cvat等)。标注人员使用图像标注工具时,先通过矩形、多边形、任意曲线等对图像中的目标实体进行框定,再通过不断的点击操作,勾勒出目标实体的轮廓线,最后为勾勒出的目标实体添加标签,即完成一次图像标注。
3、相关技术中,标注人员需执行的标注操作较为繁琐,标注效率低下。
技术实现思路
1、本申请提供了一种图像标注方法、装置、设备及介质,提供了一种自动预测出图像中的目标实体的方案。所述技术方案包括如下内容。
2、根据本申请的一个方面,提供了一种图像标注方法,方法包括:
3、显示当前图像;响应于接收到对所述当前图像上的目标实体的点击操作,得到点击点的点击坐标,所述目标实体是待标注的实体;
4、获取所述目标实体在所述当前图像上所处的第一图像区域的遮罩图像,以及所述第一图像区域的轮廓坐标,所述第一图像区域是所述当前图像上的部分区域;
5、基于所述当前图像、所述第一图像区域的遮罩图像、所述点击坐标以及所述第一图像区域的轮廓坐标,预测得到目标实体的遮罩图像;
6、将所述目标实体的遮罩图像添加至所述当前图像,得到并显示出处理后的图像;接收标签输入操作,对所述目标实体进行标注。
7、根据本申请的另一方面,提供了一种图像标注装置,所述装置包括:
8、显示模块,用于显示当前图像;
9、获取模块,用于响应于接收到对所述当前图像上的目标实体的点击操作,得到点击点的点击坐标,所述目标实体是待标注的实体;
10、所述获取模块,还用于获取所述目标实体在所述当前图像上所处的第一图像区域的遮罩图像,以及所述第一图像区域的轮廓坐标,所述第一图像区域是所述当前图像上的部分区域;
11、预测模块,用于基于所述当前图像、所述第一图像区域的遮罩图像、所述点击坐标以及所述第一图像区域的轮廓坐标,预测得到目标实体的遮罩图像;
12、所述显示模块,用于将所述目标实体的遮罩图像添加至所述当前图像,得到并显示出处理后的图像;
13、所述获取模块,还用于接收标签输入操作;所述显示模块,还用于对所述目标实体进行标注。
14、根据本申请的一个方面,提供了一种图像标注工具,所述图像标注工具包括计算机指令,所述计算机指令存储在计算机可读存储介质中,所述计算机指令由处理器加载并执行以实现如上的图像标注方法。
15、根据本申请的一个方面,提供了一种计算机设备,计算机设备包括:处理器和存储器,存储器存储有计算机程序,计算机程序由处理器加载并执行以实现如上的图像标注方法。
16、根据本申请的另一方面,提供了一种计算机可读存储介质,存储介质存储有计算机程序,计算机程序由处理器加载并执行以实现如上的图像标注方法。
17、根据本申请的另一个方面,提供了一种计算机程序产品或计算机程序,该计算机程序产品或计算机程序包括计算机指令,该计算机指令存储在计算机可读存储介质中。计算机设备的处理器从计算机可读存储介质读取该计算机指令,处理器执行该计算机指令,使得该计算机设备执行上述图像标注方法。
18、本申请实施例提供的技术方案带来的有益效果至少包括如下内容。
19、通过当前图像、当前图像上的目标实体的点击操作的点击坐标、第一图像区域(当前图像上包含目标实体的部分区域)的遮罩图像、第一图像区域的轮廓坐标,共四种输入,预测得到目标实体的遮罩图像。将目标实体的遮罩图像添加至当前图像上,当前图像可显示出分割得到的目标实体,再根据标签即可对目标实体进行标注。
20、本申请提供了利用四种输入自动预测得到当前图像上的目标实体的方案。相关技术中,需要由用户不断的点击操作以勾勒出目标实体,本申请将通过ai预测的方式分割出目标实体,ai预测仅需要用户少量的点击操作即可预测目标实体,大大减少了用户的点击操作次数,提升了标注效率。并且,相比于用户手动勾勒出的目标实体,ai预测得到的目标实体更准确。
1.一种图像标注方法,其特征在于,所述方法包括:
2.根据权利要求1所述的方法,其特征在于,所述基于所述当前图像、所述第一图像区域的遮罩图像、所述点击坐标以及所述第一图像区域的轮廓坐标,预测得到目标实体在所述当前图像上所处的目标实体的遮罩图像,包括:
3.根据权利要求2所述的方法,其特征在于,所述对所述点击坐标进行编码,得到第四特征向量,包括:
4.根据权利要求2所述的方法,其特征在于,所述第一图像区域的轮廓坐标包括所述第一图像区域的多个轮廓点的坐标;
5.根据权利要求2至4任一所述的方法,其特征在于,所述点击点为第一点;所述方法还包括:
6.根据权利要求2至4任一所述的方法,其特征在于,所述方法还包括:
7.根据权利要求2至4任一所述的方法,其特征在于,所述通过所述遮罩解码器输出所述目标实体的遮罩图像,包括:
8.根据权利要求7所述的方法,其特征在于,所述通过所述遮罩解码器输出多个候选遮罩图像,包括:
9.根据权利要求1至4任一所述的方法,其特征在于,
10.根据权利要求9所述的方法,其特征在于,所述获取所述目标实体在所述当前图像上所处的第一图像区域的遮罩图像,以及所述第一图像区域的轮廓坐标,包括:
11.一种图像标注装置,其特征在于,所述装置包括:
12.一种图像标注工具,其特征在于,所述图像标注工具包括计算机指令,所述计算机指令存储在计算机可读存储介质中,所述计算机指令由处理器加载并执行以实现如权利要求1至10任一所述的图像标注方法。
13.一种计算机设备,其特征在于,所述计算机设备包括:处理器和存储器,所述存储器存储有计算机程序,所述计算机程序由所述处理器加载并执行以实现如权利要求1至10任一所述的图像标注方法。
14.一种计算机可读存储介质,其特征在于,所述计算机可读存储介质存储有计算机程序,所述计算机程序由处理器加载并执行以实现如权利要求1至10任一所述的图像标注方法。
15.一种计算机程序产品,其特征在于,所述计算机程序产品包括计算机指令,所述计算机指令存储在计算机可读存储介质中,所述计算机指令由处理器加载并执行以实现如权利要求1至10任一所述的图像标注方法。