本发明涉及图像处理技术领域,尤其涉及一种图像数据标注色斑自动修正的方法。
背景技术:
图像数据标注过程是通过photoshop等抠图软件扣出的,由于抠图的两个对象紧密接触时,抠出的图会存在少量不属于任何一个对象的像素,即色斑。这些像素肉眼无法识别,但是却对机器学习模型的训练造成非常大的影响。
技术实现要素:
本发明所要解决的技术问题是克服现有技术中存在的不足,提供一种能够将色斑去除的图像数据标注色斑自动修正的方法。
本发明是通过以下技术方案予以实现:
一种图像数据标注色斑自动修正的方法,其特征在于,包括以下步骤:
a.读取图像,获得图像对应的索引值矩阵及颜色映射矩阵;
b.通过连通图识别的方法,得到步骤a的索引值矩阵中不同索引值对应的图像块;
c.识别图像块的像素数目,如果像素数目小于设定值,则将该图像块对应的索引值替换为临近该图像块的像素的索引值。
优选的,设定值为100。
本发明的有益效果是:
本发明将图片读取成索引值和颜色映射矩阵,通过连通图识别的方法将少量不属于任何对象的像素识别出来,并设置为邻近像素的索引值,该方法弥补了使用抠图软件等工具无法查出色斑问题的困难,同时将标注员的精力集中在抠图任务上,提升了标注效率。
附图说明
图1是本发明的流程示意图。
具体实施方式
为了使本技术领域的技术人员更好地理解本发明的技术方案,下面结合附图和最佳实施例对本发明作进一步的详细说明。
如图所示,本发明包括以下步骤:
a.读取图像,获得图像对应的索引值矩阵及颜色映射矩阵,索引值矩阵为像素点对应颜色的索引值形成的矩阵,任一颜色的索引值等于该颜色在颜色映射矩阵中所在行数的数值减一;
b.通过连通图识别的方法,得到步骤a的索引值矩阵中不同索引值对应的图像块;
c.分别识别步骤b中得到的多个图像块的像素数目,如果像素数目小于设定值,则将该图像块对应的索引值替换为临近该图像块的像素的索引值。
优选的,设定值为100,通常色斑的像素数小于100,通过本方法即可完成对图像中色斑的自动修正。
以上所述仅为本发明的优选实施例而已,并不用于限制本发明,对于本领域的技术人员来说,本发明可以有各种更改和变化。凡在本发明的精神和原则之内,所作的任何修改、等同替换、改进等,均应包含在本发明的保护范围之内。